ByteDance-Seed / ByteCheckpointLinks
ByteCheckpoint: An Unified Checkpointing Library for LFMs
☆261Updated last month
Alternatives and similar repositories for ByteCheckpoint
Users that are interested in ByteCheckpoint are comparing it to the libraries listed below
Sorting:
- Accelerating MoE with IO and Tile-aware Optimizations☆542Updated last week
- Bridge Megatron-Core to Hugging Face/Reinforcement Learning☆183Updated last week
- Allow torch tensor memory to be released and resumed later☆202Updated last week
- Zero Bubble Pipeline Parallelism☆447Updated 8 months ago
- ☆340Updated 2 weeks ago
- Training library for Megatron-based models with bi-directional Hugging Face conversion capability☆363Updated this week
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆221Updated last year
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆365Updated 6 months ago
- Pipeline Parallelism Emulation and Visualization☆75Updated last week
- PyTorch bindings for CUTLASS grouped GEMM.☆182Updated last month
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆283Updated 2 months ago
- Toolchain built around the Megatron-LM for Distributed Training☆84Updated last month
- Distributed MoE in a Single Kernel [NeurIPS '25]☆183Updated this week
- Miles is an enterprise-facing reinforcement learning framework for large-scale MoE post-training and production workloads, forked from an…☆744Updated this week
- Autonomous GPU Kernel Generation via Deep Agents☆217Updated this week
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆160Updated 3 months ago
- Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.☆152Updated last month
- ☆444Updated 5 months ago
- Utility scripts for PyTorch (e.g. Make Perfetto show some disappearing kernels, Memory profiler that understands more low-level allocatio…☆80Updated 4 months ago
- A NCCL extension library, designed to efficiently offload GPU memory allocated by the NCCL communication library.☆80Updated last month
- SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs☆187Updated 3 months ago
- 16-fold memory access reduction with nearly no loss☆109Updated 9 months ago
- A simple calculation for LLM MFU.☆64Updated 4 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆140Updated 7 months ago
- Genai-bench is a powerful benchmark tool designed for comprehensive token-level performance evaluation of large language model (LLM) serv…☆251Updated this week
- nnScaler: Compiling DNN models for Parallel Training☆123Updated 3 months ago
- Train speculative decoding models effortlessly and port them smoothly to SGLang serving.☆639Updated this week
- [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank☆67Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆626Updated this week
- JAX backend for SGL☆227Updated this week