A fast MoE impl for PyTorch
Role in this project:
ML Engineer Contributions:7 releases, 57 reviews, 300 commits in 2 years
Contributions summary:Rick's contributions primarily involve implementing and optimizing a fast MoE (Mixture of Experts) implementation for PyTorch. Their work includes creating CUDA kernels for batched matrix multiplication, designing a global exchange mechanism, and building kernels for optimized scatter and gather operations. They have added code for performance testing and incorporated a testing infrastructure within the repository to validate and benchmark their CUDA-based MoE layer.
pytorchmoemixture-of-expertsimplpytorch-lightning
Contributions:142 commits, 3 PRs, 94 pushes in 2 months
discovernew-worldlet