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