patrickvonplaten/optax
Optax is a gradient processing and optimization library for JAX.
PythonApache-2.0
Watchers
No one’s watching this repository yet.
Optax is a gradient processing and optimization library for JAX.
PythonApache-2.0
No one’s watching this repository yet.