patrickvonplaten/optax
Optax is a gradient processing and optimization library for JAX.
PythonApache-2.0
No issues in this repository yet.
Optax is a gradient processing and optimization library for JAX.
PythonApache-2.0
No issues in this repository yet.