jax-ml/jax

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

[view on github]last commit: Oct 20, 2026

stars

36,471

7d

+59

30d

+154

90d

+462

## star history

## found in

Awesome Open Source AI/🧬 1. Core Frameworks & Libraries