← all repositories

laekov/fastmoe

A PyTorch extension providing fast and efficient Mixture of Experts (MoE) implementation for training large neural network models.

1.9k stars Python ML FrameworksLanguage Models
fastmoe
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

FastMoE provides an easy-to-use system for supporting Mixture of Experts models in PyTorch. It contains customized C and Python operators for efficient distributed training of MoE layers. The project enables extending Transformer-based models with MoE architecture through a simple plugin interface, supporting features like distributed experts and NCCL-based communication.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.