Add NorMuon optimizer#1612

tinker495 wants to merge 1 commit intogoogle-deepmind:maingoogle-deepmind/optax:mainfrom

tinker495:feature/add-normuontinker495/optax:feature/add-normuonCopy head branch name to clipboard