docs: Grokfast

kozistr · kozistr · commit 7a39a1eb3ac4 · 2024-06-22T14:57:00.000+09:00
diff --git a/docs/optimizer.md b/docs/optimizer.md
@@ -156,6 +156,18 @@
     :docstring:
     :members:
 
+::: pytorch_optimizer.gradfilter_ema
+    :docstring:
+    :members:
+
+::: pytorch_optimizer.gradfilter_ma
+    :docstring:
+    :members:
+
+::: pytorch_optimizer.GrokFastAdamW
+    :docstring:
+    :members:
+
 ::: pytorch_optimizer.GSAM
     :docstring:
     :members:
