google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.
1,828Updated last week

Alternatives and similar repositories for optax:

Users that are interested in optax are comparing it to the libraries listed below