Illustration: the project’s public image, pixelated by RepoMasters. Credits
RCEpic

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

Owner
google-deepmind (organization)
Stars
Stars: 2,360
Forks
Forks: 388
Language
Python
License
Apache-2.0
Created
2020
Topics
machine-learningoptimization
Tier
RC (2,000 to 4,999 ★) What this tier means
View on GitHub

More repos by google-deepmind

All 174 repos by google-deepmind

Nearby in Python

All Python repos