NVIDIA-NeMo / Megatron-BridgeLinks
Training library for Megatron-based models
☆74Updated this week
Alternatives and similar repositories for Megatron-Bridge
Users that are interested in Megatron-Bridge are comparing it to the libraries listed below
Sorting:
- Odysseus: Playground of LLM Sequence Parallelism☆77Updated last year
- ☆126Updated 3 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆216Updated last year
- A lightweight reinforcement learning framework that integrates seamlessly into your codebase, empowering developers to focus on algorithm…☆68Updated 3 weeks ago
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆43Updated 2 months ago
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆207Updated last week
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆127Updated 9 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆239Updated last month
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆185Updated 3 months ago
- ☆111Updated last year
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆110Updated 6 months ago
- ☆142Updated 7 months ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆134Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆82Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆119Updated 3 months ago
- ☆96Updated 4 months ago
- Megatron's multi-modal data loader☆243Updated 2 weeks ago
- ☆121Updated last year
- Toolchain built around the Megatron-LM for Distributed Training☆65Updated 2 weeks ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆40Updated last year
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆141Updated 4 months ago
- Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.☆89Updated last week
- Triton-based implementation of Sparse Mixture of Experts.☆239Updated 3 weeks ago
- [NeurIPS 2024] The official implementation of "Kangaroo: Lossless Self-Speculative Decoding for Accelerating LLMs via Double Early Exitin…☆60Updated last year
- Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"☆120Updated last year
- Distributed IO-aware Attention algorithm☆21Updated 2 weeks ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆168Updated last year
- ☆95Updated 5 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆106Updated 5 months ago
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆48Updated 11 months ago