2 Commits

Author	SHA1	Message	Date
Wing Lian	6130e40c37	fix flaky tests; should be using train loss from final step rather than final avg train loss	2026-03-22 10:38:46 -04: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