lucidrains / soft-moe-pytorch

Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch
MIT License
239 stars 8 forks source link