microsoft / Tutel
Tutel MoE: An Optimized Mixture-of-Experts Implementation
☆735Updated this week
Related projects ⓘ
Alternatives and complementary repositories for Tutel
- Microsoft Automatic Mixed Precision Library☆525Updated last month
- A fast MoE impl for PyTorch☆1,565Updated 4 months ago
- ☆574Updated last week
- A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models☆642Updated last year
- Ring attention implementation with flash attention☆585Updated last week
- A curated reading list of research in Mixture-of-Experts(MoE).☆538Updated 3 weeks ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,338Updated 8 months ago
- Pipeline Parallelism for PyTorch☆726Updated 2 months ago
- Transformers with Arbitrarily Large Context☆641Updated 3 months ago
- Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch☆293Updated 5 months ago
- A collection of AWESOME things about mixture-of-experts☆972Updated 3 months ago
- Code for the ALiBi method for transformer language models (ICLR 2022)☆507Updated last year
- Zero Bubble Pipeline Parallelism☆281Updated last week
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,893Updated last month
- [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.☆391Updated 3 months ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,257Updated 4 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆357Updated this week
- Best practice for training LLaMA models in Megatron-LM☆628Updated 10 months ago
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,149Updated last month
- PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538☆983Updated 7 months ago
- ☆289Updated 7 months ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆476Updated 3 weeks ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆195Updated 3 months ago
- Official implementation of TransNormerLLM: A Faster and Better LLM☆229Updated 9 months ago
- Helpful tools and examples for working with flex-attention☆469Updated 3 weeks ago
- LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training☆390Updated this week
- A simple and effective LLM pruning approach.☆669Updated 3 months ago
- Code for the ICML 2023 paper "SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shot".☆721Updated 3 months ago
- Latency and Memory Analysis of Transformer Models for Training and Inference☆355Updated last week