optax
PublicOptax is a gradient processing and optimization library for JAX.
Creat:2020-06-12T23:45:35
Update:2025-03-27T04:14:44
https://optax.readthedocs.io
2.0K
Stars
2
Stars Increase
Optax is a gradient processing and optimization library for JAX.