Research Paper Deep Dive - The Sparsely-Gated Mixture-of-Experts (MoE) 

can i used 3D decoder Unet instead of MLP model as expert , which means do i need to modify the full Framwork introduced in S-MoE and Gate netwoek i need to make able to process 3D imagfe to compute Wieght and topK route