jax-ml/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

+746

## star history

## found in

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