Refactor Muon optimizer constructors#6147

janEbert wants to merge 5 commits into

NVIDIA:mainNVIDIA/Megatron-LM:mainfrom

janEbert:muon-init-refactorjanEbert/Megatron-LM:muon-init-refactorCopy head branch name to clipboard