issues
search
google-deepmind
/
optax
Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.63k
stars
174
forks
source link
Detail SGD with momentum equations.
#830
Closed
copybara-service[bot]
closed
6 months ago
copybara-service[bot]
commented
6 months ago
Detail SGD with momentum equations.
Detail SGD with momentum equations.