The algorithm works for MoE load balancing in general