[MoE] Add optional output normalization for latent MoE - #6449

[MoE] Add optional output normalization for latent MoE#6449

86MaxCao wants to merge 1 commit into

NVIDIA:devNVIDIA/Megatron-LM:devfrom

86MaxCao:k3-latent-moe-output-norm86MaxCao/Megatron-LM:k3-latent-moe-output-normCopy head branch name to clipboard