Adobe Interview Question

Implement MoE for a single transformer layer