google/jax

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

[view on github]last commit: Jun 19, 2026

stars

35,852

7d

+46

30d

+208

90d

+748

## star history

## found in

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