Triton-based implementation of Sparse Mixture of Experts. - View it on GitHub