jax-ml / jax

git clone https://github.com/jax-ml/jax.git

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

stars, 24h · #5,491

+~30

stars, 7 days · #6,149

+~130

stars, 30 days · #5,771

36k

★ total · +0% this week

3.9k

forks · 2.6k open issues

2026-10-20 · 36,474 ★ 2026-10-23 · 36,487 ★

LanguagePython

LicenseApache-2.0

Created2018-10-25 (7y ago)

Last pushtoday

OwnerOrganization

jax