Refactor of pre/post weight update hook functions#224

mkhona-nvidia wants to merge 2 commits into

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

mkhona-nvidia:mkhona/pre_post_weight_update_refactorsmkhona-nvidia/Emerging-Optimizers:mkhona/pre_post_weight_update_refactorsCopy head branch name to clipboard