Add CAME Optimizer (#2385)

2025-05-07 10:31:46 -04:00
parent 3c8b9b33d6
commit f943306263
6 changed files with 66 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@@ -142,6 +142,7 @@ extras_require = {
        "apollo-torch",
        "lomo-optim==0.1.1",
        "torch-optimi==0.2.1",
+        "came_pytorch==0.1.3",
    ],
    "ray": [
        "ray[train]",