replace router_aux_loss_coef with moe_loss_weight 500c7b3 verified eitanturok commited on Apr 15, 2024