jax

Name: jax
Rating: 5 (35401 reviews)

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

github Developer Tools Python free

★ 35,401Stars

3,528Forks

35,401Watchers

3Views

Apr 2026Last Update

View on GitHub →

About jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more