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

• stanford-futuredata / stk
  ☆112Updated last year
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆216Updated last year
• anyscale / llm-continuous-batching-benchmarks
  ☆121Updated last year
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆246Updated 3 weeks ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆78Updated last year
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆120Updated 11 months ago
• zhengzangw / Sequence-Scheduling
  PyTorch implementation of paper "Response Length Perception and Sequence Scheduling: An LLM-Empowered LLM Inference Pipeline".
  ☆92Updated 2 years ago
• ISEEKYAN / mbridge
  Bridge Megatron-Core to Hugging Face/Reinforcement Learning
  ☆143Updated this week
• FasterDecoding / REST
  REST: Retrieval-Based Speculative Decoding, NAACL 2024
  ☆210Updated last month
• cli99 / llm-analysis
  Latency and Memory Analysis of Transformer Models for Training and Inference
  ☆460Updated 6 months ago
• teelinsan / parallel-decoding
  Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"
  ☆120Updated last year
• dilab-zju / self-speculative-decoding
  Code associated with the paper **Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding**
  ☆206Updated 8 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆125Updated 5 months ago
• haochengxi / Train_Transformers_with_INT4
  ☆156Updated 2 years ago
• epfml / dynamic-sparse-flash-attention
  ☆149Updated 2 years ago
• Dao-AILab / grouped-latent-attention
  ☆130Updated 5 months ago
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆283Updated last year
• NVIDIA-NeMo / Megatron-Bridge
  Training library for Megatron-based models
  ☆142Updated this week
• lucidrains / speculative-decoding
  Explorations into some recent techniques surrounding speculative decoding
  ☆288Updated 10 months ago
• thu-ml / low-bit-optimizers
  Low-bit optimizers for PyTorch
  ☆132Updated 2 years ago
• thu-pacman / FasterMoE
  ☆87Updated 3 years ago
• thunlp / Ouroboros
  Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)
  ☆111Updated 7 months ago
• alexzhang13 / flashattention2-custom-mask
  Triton implementation of FlashAttention2 that adds Custom Masks.
  ☆142Updated last year
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆161Updated 3 weeks ago
• shreyansh26 / Speculative-Sampling
  Implementation of Speculative Sampling as described in "Accelerating Large Language Model Decoding with Speculative Sampling" by Deepmind
  ☆104Updated last year
• sail-sg / zero-bubble-pipeline-parallelism
  Zero Bubble Pipeline Parallelism
  ☆433Updated 5 months ago
• opengear-project / GEAR
  GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM
  ☆169Updated last year
• fzyzcjy / torch_utils
  Utility scripts for PyTorch (e.g. Memory profiler that understands more low-level allocations such as NCCL)
  ☆62Updated last month
• rlite-project / RLite
  A lightweight reinforcement learning framework that integrates seamlessly into your codebase, empowering developers to focus on algorithm…
  ☆68Updated 2 months ago
• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆48Updated last month