added MoE + Trainer