- Repository
- https://github.com/jax-ml/jax
- Homepage
- https://docs.jax.dev
- Owner
- jax-ml (Organization)
- Stars
- 36,463
- Forks
- 3,867
- Open issues
- 2,571
- Language
- Python
- License
- Apache-2.0
- Created
- 2018-10-25
- Last push
- 2026-10-18
- Topics
- jax
- Snapshot
- 2026-10-18
GitHub · #194
jax-ml/jax
jax-ml/jax is #194 in this GitHub directory, with 36,463 stars. Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more