// Official vendor repos
JAX
4 tracked official repos · 2 releases in the last 30 days.
01Latest releases
-
jax-ml/jax jax-v0.12.0 Minor
New features
-
jax-ml/jax-triton v0.4.1 Patch
jax-triton contains integrations between JAX and OpenAI Triton
02All repos
| Repo | Category | ★ | Latest | Licence |
|---|---|---|---|---|
| jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more | Training & fine-tuning | 36.5k | jax-v0.12.02026-10-19 | Apache-2.0 |
| scaling-bookHome for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs | Learning & awesome lists | 1,491 | — | MIT |
| jax-tritonjax-triton contains integrations between JAX and OpenAI Triton | Kernels, compilers & acceleration | 472 | v0.4.12026-09-24 | Apache-2.0 |
| ml_dtypesA stand-alone implementation of several NumPy dtype extensions used in machine learning. | Kernels, compilers & acceleration | 363 | v0.6.02026-08-13 | Apache-2.0 |