MemFine: Memory-Aware Fine-Grained Scheduling for MoE Training

Lu Zhao, Rong Shi, Shaoqing Zhang +21

The training of large-scale Mixture of Experts (MoE) models faces a critical memory bottleneck due to severe load imbalance caused by dynamic token routing. This imbalance leads to…