Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch
Full Changelog: https://github.com/lucidrains/soft-moe-pytorch/compare/0.1.5...0.1.6