microsoft / Tutel
Tutel MoE: Optimized Mixture-of-Experts Library, Support DeepSeek FP8/FP4
☆820Updated this week
Alternatives and similar repositories for Tutel
Users that are interested in Tutel are comparing it to the libraries listed below
Sorting:
- Microsoft Automatic Mixed Precision Library☆595Updated 7 months ago
- A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models☆739Updated last year
- Large Context Attention☆710Updated 3 months ago
- Ring attention implementation with flash attention☆759Updated last month
- Pipeline Parallelism for PyTorch☆764Updated 8 months ago
- A fast MoE impl for PyTorch☆1,716Updated 3 months ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,401Updated 10 months ago
- ☆631Updated 2 weeks ago
- A curated reading list of research in Mixture-of-Experts(MoE).☆623Updated 6 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,386Updated last year
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,246Updated 2 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆2,066Updated last month
- PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538☆1,103Updated last year
- Zero Bubble Pipeline Parallelism☆389Updated last week
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,566Updated last year
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆512Updated 6 months ago
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…☆2,400Updated this week
- Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch☆330Updated 10 months ago
- Helpful tools and examples for working with flex-attention☆766Updated last week
- A collection of AWESOME things about mixture-of-experts☆1,108Updated 5 months ago
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆608Updated last year
- Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".☆2,101Updated last year
- Best practice for training LLaMA models in Megatron-LM☆650Updated last year
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆808Updated this week
- Latency and Memory Analysis of Transformer Models for Training and Inference☆409Updated 3 weeks ago
- ☆319Updated last year
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆284Updated 2 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆492Updated 3 weeks ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆377Updated last year
- [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.☆444Updated 9 months ago