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.
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-06 | 269 |
Today
— stars today
This week
— stars this week
This month
— stars this month
Momentum
0.0
growth rate 0.00%/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'