google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.

https://optax.readthedocs.io

Apache License 2.0

1.65k stars 181 forks source link

Add a nesterov flag to radam optimizer. #949

Closed carlosgmartin closed 5 months ago

carlosgmartin commented 5 months ago

944