jax-ml/jax

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

[view on github]last commit: May 24, 2026

stars

35,684

7d

+55

30d

+202

90d

+826

## star history

## found in

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