/pytorch-mixtures

One-stop solutions for Mixture of Experts and Mixture of Depth modules in PyTorch.

Primary LanguagePythonMIT LicenseMIT

Stargazers