google-deepmind / optaxLinks

Optax is a gradient processing and optimization library for JAX.
1,959Updated this week

Alternatives and similar repositories for optax

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

Sorting: