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