Alternatives and similar repositories for Dive-into-Big-Model-Training

Users that are interested in Dive-into-Big-Model-Training are comparing it to the libraries listed below

• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆216Updated last year
• stanford-futuredata / stk
  ☆111Updated last year
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆121Updated 9 months ago
• anyscale / llm-continuous-batching-benchmarks
  ☆121Updated last year
• ISEEKYAN / mbridge
  Bridge Megatron-Core to Hugging Face/Reinforcement Learning
  ☆120Updated 2 weeks ago
• yanring / Megatron-MoE-ModelZoo
  Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.
  ☆89Updated last week
• NVIDIA-NeMo / Megatron-Bridge
  Training library for Megatron-based models
  ☆74Updated this week
• cli99 / llm-analysis
  Latency and Memory Analysis of Transformer Models for Training and Inference
  ☆454Updated 5 months ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆77Updated last year
• thu-pacman / FasterMoE
  ☆86Updated 3 years ago
• FasterDecoding / REST
  REST: Retrieval-Based Speculative Decoding, NAACL 2024
  ☆207Updated last week
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆239Updated 3 weeks ago
• yaof20 / Flash-RL
  Implementation for FP8/INT8 Rollout for RL training without performence drop.
  ☆200Updated this week
• teelinsan / parallel-decoding
  Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"
  ☆120Updated last year
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆145Updated 3 weeks ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆134Updated last year
• Hsword / Hetu
  A high-performance distributed deep learning system targeting large-scale and automated distributed training. If you have any interests, …
  ☆121Updated last year
• kssteven418 / LTP
  [KDD'22] Learned Token Pruning for Transformers
  ☆100Updated 2 years ago
• epfml / dynamic-sparse-flash-attention
  ☆149Updated 2 years ago
• thunlp / Ouroboros
  Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)
  ☆110Updated 6 months ago
• kssteven418 / BigLittleDecoder
  [NeurIPS'23] Speculative Decoding with Big Little Decoder
  ☆94Updated last year
• hpcaitech / PaLM-colossalai
  Scalable PaLM implementation of PyTorch
  ☆190Updated 2 years ago
• haochengxi / Train_Transformers_with_INT4
  ☆154Updated 2 years ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆54Updated last year
• lucidrains / speculative-decoding
  Explorations into some recent techniques surrounding speculative decoding
  ☆285Updated 8 months ago
• Ascend / AscendSpeed
  ☆79Updated last year
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆211Updated this week
• sail-sg / zero-bubble-pipeline-parallelism
  Zero Bubble Pipeline Parallelism
  ☆426Updated 4 months ago
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆278Updated last year
• Dao-AILab / grouped-latent-attention
  ☆126Updated 3 months ago