Optax is a gradient processing and optimization library for JAX.
Python1144apache-2.0
10 months ago
Python1364apache-2.0
3 months ago