Commit Graph

4 Commits

Author SHA1 Message Date
Wing Lian
f77408a3d0 fix tests 2026-04-23 23:47:28 +00:00
Wing Lian
5db4272f69 more steps for loss check 2026-04-23 18:43:18 +00:00
Wing Lian
431888c1de use smaller pretrained models for ci 2026-04-23 13:51:01 +00:00
salman
bbd3486f57 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 10:43:47 -05:00