// Train & fine-tune
jaxjax-v0.11.2
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Open source Maintained For developers
IN ONE PARAGRAPH
jax is composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.
Good for: Developers and teams who want to train & fine-tune.
01How to start
- 1Read the official quick-start (link above).
- 2Install from the official package registry:
pip install jax - 3Try it on a small example before wiring it into a project.
02Alternatives & comparison
Compare side by side →Versions, why it is listed, for AI to read
Recent releases
- jax-v0.11.2 2026-09-17
- jax-v0.11.1 2026-08-17
- jax-v0.11.0 2026-07-16
- jax-v0.10.2 2026-06-17
- jax-v0.10.1 2026-05-20
For AI agents
- Interfaces:SDK
- Machine-readable record:/api/v1/software/jax.json · llms.txt
- Added 2026-10-10 · updated 2026-10-22