jax-ml/jax

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

[view on github]last commit: Apr 3, 2026

stars

35,295

7d

-

30d

-

90d

-

## star history

## found in

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