fix(muon_utils): keep newton_schulz scale-invariant for small-norm inputs (#229)#230

skyw merged 3 commits into

NVIDIA-NeMo:mainNVIDIA-NeMo/Emerging-Optimizers:mainfrom

yuchenwang3:fix/newton-schulz-eps-scale-invarianceyuchenwang3/Emerging-Optimizers:fix/newton-schulz-eps-scale-invarianceCopy head branch name to clipboard