jax-ml/jax

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

Language: Python

36,463GitHub stars

3,867public forks

Observed star growth

+3 stars24h growth
Observed OpenPulse samples

Building history7d growth
Not enough current observations

Building history30d growth
Not enough current observations

Latest OpenPulse observation: 2026-10-19 01:48 UTC