feat(pt): optimze HybridMuon by borrowing some ideas from deepseek v4 paper#5424

OutisLi wants to merge 1 commit intodeepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

OutisLi:pr/muonOutisLi/deepmd-kit:pr/muonCopy head branch name to clipboard