feat: support qwen3 next with muon#2875

Prayer3th wants to merge 1 commit intoAI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

primatrix:feat/support-muonprimatrix/maxtext:feat/support-muonCopy head branch name to clipboard