Adobe interview question

Implement MoE for a single transformer layer