Add a sharding-aware version of Muon to maxtext. - #4936
Open
copybara-service[bot] wants to merge 1 commit into
Open
Add a sharding-aware version of Muon to maxtext.#4936copybara-service[bot] wants to merge 1 commit into
copybara-service[bot] wants to merge 1 commit into
Commits
Commits on Aug 20, 2026
- authored andcommitted