Add configurable Newton-Schulz iteration dtype#228

hyleepp wants to merge 1 commit into

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

hyleepp:codex/ns-dtype-fp16hyleepp/Emerging-Optimizers:codex/ns-dtype-fp16Copy head branch name to clipboard