feat(megatron): add after-train-step hook#1682

waitong94 wants to merge 2 commits into

radixark:mainradixark/miles:mainfrom

GymPod:waitong/megatron-after-train-step-hookGymPod/miles-upstream:waitong/megatron-after-train-step-hookCopy head branch name to clipboard