Docstrings for all solvers with Objective Function values

PiperOrigin-RevId: 604329001
google-deepmind · Feb 5, 2024 · d8475ff · d8475ff
1 parent 5a999ca
commit d8475ff
Show file tree

Hide file tree

Showing 2 changed files with 460 additions and 0 deletions.
diff --git a/optax/__init__.py b/optax/__init__.py
@@ -41,6 +41,7 @@
 from optax._src.alias import optimistic_gradient_descent
 from optax._src.alias import radam
 from optax._src.alias import rmsprop
+from optax._src.alias import rprop
 from optax._src.alias import sgd
 from optax._src.alias import sm3
 from optax._src.alias import yogi