Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Minimal yet performant LLM examples in pure JAX
| Date | Stars |
|---|---|
| 2026-07-31 | 269 |
| 2026-08-11 | 272 |
| 2026-08-18 | 273 |
| 2026-08-19 | 273 |
| 2026-08-20 | 274 |
| 2026-08-21 | 276 |
| 2026-08-25 | 276 |
| 2026-08-27 | 277 |
| 2026-08-29 | 277 |
| 2026-09-13 | 278 |
| 2026-09-16 | 280 |
| 2026-09-17 | 281 |
| 2026-09-20 | 282 |
Today
+1 stars today
This week
+4 stars this week
This month
+6 stars this month
Momentum
8.0
growth rate 1.44%/day
A collection (in progress) of example high-performance large language model implementations, written with JAX.
Current contents include:
For multi-host cluster setup and distributed training, see multi_host_README.md and the tpu_toolkit.sh script.
Rendered by GitHub from the current README
Read on GitHubWould you bet a product on this? Bounded 0–100 and slow moving.
matched fp:a966338f79ba40be, llm:topics: jax, llm, llm-inference; description: 'Minimal yet performant LLM examples in pure JAX'
matched fp:a966338f79ba40be, llm:topics: jax, llm, llm-inference; description: 'Minimal yet performant LLM examples in pure JAX'
matched fp:a966338f79ba40be, llm:topics: jax, llm, llm-inference; description: 'Minimal yet performant LLM examples in pure JAX'
