Add arxiv reference to schedule_free optimizer.

PiperOrigin-RevId: 649572688
google-deepmind · Jul 5, 2024 · 7f62472 · 7f62472
1 parent 99967d2
commit 7f62472
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/optax/contrib/_schedule_free.py b/optax/contrib/_schedule_free.py
@@ -97,6 +97,9 @@ def schedule_free(
   optimizer. As of Apr, 2024, schedule_free is tested with SGD and Adam.
 
   References:
+    Defazio et al, `The Road Less Scheduled
+    <https://arxiv.org/abs/2405.15682>`_, 2024
+
     Defazio et al, `Schedule-Free Learning - A New Way to Train
     <https://github.com/facebookresearch/schedule_free/tree/main>`_, 2024