JAX

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT

7d +0.10% +36 stars in 7d

30d Not tracked yet — first tracked 2026-08-13

90d Not tracked yet — first tracked 2026-08-13

36,154 → 36,217 stars since Aug 13

★ 36,212 stars on GitHub Visit site ↗