google-deepmind / optax

git clone https://github.com/google-deepmind/optax.git

Optax is a gradient processing and optimization library for JAX.

stars, 24h · #55,293

+~0

stars, 7 days · #55,367

+~0

stars, 30 days · #55,505

2.4k

★ total

383

forks · 141 open issues

2026-10-20 · 2,353 ★ 2026-10-22 · 2,353 ★

LanguagePython

LicenseApache-2.0

Created2020-06-12 (6y ago)

Last push2d ago

OwnerOrganization