Top AI Repos — open-source AI, indexed and scored
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.
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 |
| 2026-09-23 | 283 |
| 2026-09-24 | 284 |
| 2026-09-28 | 285 |
| 2026-10-07 | 286 |
| 2026-10-08 | 287 |
| 2026-10-09 | 286 |
| 2026-10-21 | 285 |
| 2026-10-26 | 286 |
| 2026-11-04 | 286 |
Today
— stars today
This week
— stars this week
This month
+1 stars this month
Momentum
0.0
growth rate 0.00%/day
# JAX LLM examples A collection (in progress) of example high-performance large language model implementations, written with JAX. Current contents include: * [DeepSeek R1](deepseek_r1_jax/) * [Llama 4](llama4/) * [Llama 3](llama3/) * [Qwen 3](qwen3/) * [Kimi K2](kimi_k2/) * [OpenAI GPT OSS](gpt_oss/) * [NVIDIA Nemotron 3 Nano](nemotron3/) * [Gemma 4](gemma4/) --- For multi-host cluster setup and distributed training, see [multi_host_README.md](./multi_host_README.md) and the [tpu_toolkit.sh script](./misc/tpu_toolkit.sh).
Excerpt of 533 characters
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'