LLaMA-MoE is an open-source project that builds mixture-of-experts language models from LLaMA through expert partitioning and continual pre-training. The repository is centered on making MoE research more accessible by offering smaller and more affordable models with only about 3.0 to 3.5 billion activated parameters, which helps reduce deployment and experimentation costs. Its architecture works by splitting LLaMA feed-forward networks into sparse experts and adding gating mechanisms so that only selected experts are activated during inference and training. The project is not just a model release, but also a research framework that includes multiple expert construction methods, several gating strategies, and tooling for continual pre-training on filtered SlimPajama-based datasets. It also emphasizes training efficiency through features such as FlashAttention-v2 integration and fast streaming dataset loading, which are important for large-scale experimentation.
Features
- Sparse MoE models with roughly 3.0 to 3.5B activated parameters
- Multiple expert construction methods for partitioning feed-forward networks
- Support for TopK noisy gating and Switch gating strategies
- Continual pre-training pipeline built around filtered SlimPajama data
- FlashAttention-v2 integration and streaming dataset loading
- Monitoring utilities for routing, loss, throughput, and model utilization