Spatial Mixture-of-Experts - Dryden, Nikoli; Hoefler, Torsten (Advances in Neural Information Processing Systems, 2022)