GitHub repository: whn09/fastmoe
Description: A fast MoE impl for PyTorch