Global-batch load balance almost free lunch to improve your MoE LLM training

Background The Mixture-of-Experts (MoEs) architecture has become a popular model-parameter-scale-up technique. Typically, one MoE layer cons