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