PyTorch implementation of moe, which stands for mixture of experts - View it on GitHub