Start draft PR for MLA support to Muon optimizer#4477

Prachi-kushwaha wants to merge 1 commit intoNVIDIA:mainNVIDIA/Megatron-LM:mainfrom

Prachi-kushwaha:MLA-support-to-muon-optimizerPrachi-kushwaha/Megatron-LM:MLA-support-to-muon-optimizerCopy head branch name to clipboard