Distributed Muon Optimizer (#3264)

* init * working * updating configs * removing unneeded files * lint * comments * lint * fix regex match * bump contribs version * comments * fixing tests and imports * muon imports in test v2 * test cleanup * bump contribs version --------- Co-authored-by: Salman Mohammadi <“salman.mohammadi@outlook.com”>
2025-12-19 16:43:47 +01:00
parent 3750d7dd64
commit bbd3486f57
9 changed files with 387 additions and 55 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -67,8 +67,7 @@ openenv-core==0.1.0
 schedulefree==1.4.1

 axolotl-contribs-lgpl==0.0.7
-axolotl-contribs-mit==0.0.5
-
+axolotl-contribs-mit==0.0.6
 # telemetry
 posthog==6.7.11