google/jax

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

[view on github]last commit: Apr 2, 2026

stars

35,286

7d

+88

30d

+284

90d

-

## star history

## found in

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