google/jax

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

[view on github]last commit: May 24, 2026

stars

35,685

7d

+56

30d

+203

90d

+829

## star history

## found in

awesome-machine-learning/Python awesome-machine-learning/Python