jaxplore
resources
notes
blog
about
Optax
Home
Resources
Web
Official Docs
Optax
Optax
Optax is a gradient processing and optimization library for JAX.
Copied!
Copy code to clipboard