google-deepmind / optaxLinks

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

Alternatives and similar repositories for optax

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

Sorting: