sail-sg / zero-bubble-pipeline-parallelismView external linksLinks
Zero Bubble Pipeline Parallelism
☆449May 7, 2025Updated 9 months ago
Alternatives and similar repositories for zero-bubble-pipeline-parallelism
Users that are interested in zero-bubble-pipeline-parallelism are comparing it to the libraries listed below
Sorting:
- Ring attention implementation with flash attention☆980Sep 10, 2025Updated 5 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆643Jan 15, 2026Updated 3 weeks ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Aug 19, 2024Updated last year
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,247Aug 28, 2025Updated 5 months ago
- Distributed Compiler based on Triton for Parallel Systems☆1,350Updated this week
- Byted PyTorch Distributed for Hyperscale Training of LLMs and RLs☆926Nov 27, 2025Updated 2 months ago
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on H…☆3,152Feb 7, 2026Updated last week
- Vocabulary Parallelism☆25Mar 10, 2025Updated 11 months ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆70Mar 20, 2025Updated 10 months ago
- Sequence-level 1F1B schedule for LLMs.☆38Aug 26, 2025Updated 5 months ago
- A bidirectional pipeline parallelism algorithm for computation-communication overlap in DeepSeek V3/R1 training.☆2,919Jan 14, 2026Updated last month
- Pipeline Parallelism for PyTorch☆785Aug 21, 2024Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆184Dec 16, 2025Updated last month
- LLM training technologies developed by kwai☆70Jan 21, 2026Updated 3 weeks ago
- ☆77May 4, 2021Updated 4 years ago
- FlashInfer: Kernel Library for LLM Serving☆4,935Updated this week
- Pipeline Parallelism Emulation and Visualization☆77Jan 8, 2026Updated last month
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆2,224Aug 14, 2025Updated 5 months ago
- Microsoft Automatic Mixed Precision Library☆635Dec 1, 2025Updated 2 months ago
- Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training☆1,859Feb 7, 2026Updated last week
- Ongoing research training transformer models at scale☆15,162Updated this week
- Best practice for training LLaMA models in Megatron-LM☆664Jan 2, 2024Updated 2 years ago
- A lightweight design for computation-communication overlap.☆219Jan 20, 2026Updated 3 weeks ago
- Training and serving large-scale neural networks with auto parallelization.☆3,180Dec 9, 2023Updated 2 years ago
- Estimate MFU for DeepSeekV3☆26Jan 5, 2025Updated last year
- Analyze computation-communication overlap in V3/R1.☆1,141Mar 21, 2025Updated 10 months ago
- A Easy-to-understand TensorOp Matmul Tutorial☆410Updated this week
- Perplexity GPU Kernels☆560Nov 7, 2025Updated 3 months ago
- A PyTorch native platform for training generative AI models☆5,045Updated this week
- The official repo of Pai-Megatron-Patch for LLM & VLM large scale training developed by Alibaba Cloud.☆1,527Dec 15, 2025Updated last month
- Fast low-bit matmul kernels in Triton☆429Feb 1, 2026Updated last week
- Large Context Attention☆766Oct 13, 2025Updated 4 months ago
- A high-performance distributed deep learning system targeting large-scale and automated distributed training.☆333Dec 13, 2025Updated 2 months ago
- Ongoing research training transformer models at scale☆18Feb 5, 2026Updated last week
- InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencie…☆418Aug 21, 2025Updated 5 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆265Oct 3, 2025Updated 4 months ago
- A throughput-oriented high-performance serving framework for LLMs☆945Oct 29, 2025Updated 3 months ago
- nnScaler: Compiling DNN models for Parallel Training☆124Sep 23, 2025Updated 4 months ago
- Mooncake is the serving platform for Kimi, a leading LLM service provided by Moonshot AI.☆4,701Updated this week