Optax is a gradient processing and optimization library for JAX. - View it on GitHub
Star
1951
Rank
19548