/soft-moe-pytorch

Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch

Primary LanguagePythonMIT LicenseMIT

Stargazers