jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
★ 35,401Stars
3,528Forks
35,401Watchers
3Views
Apr 2026Last Update
About jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more