Add CAME Optimizer (#2385)

2025-05-07 10:31:46 -04:00
parent 32f51bca35
commit 25e6c5f9bd
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]",