Trainer for MoE on crafted features