Stochastic Top-K Expert Routing for Mixture-of-Experts Networks - Coding problem | OneNoughtOne