tomweingarten / optax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
Home Page: https://optax.readthedocs.io
License: Apache License 2.0