Alternatives and similar repositories for Megatron-DeepSpeed

Users that are interested in Megatron-DeepSpeed are comparing it to the libraries listed below

• deepspeedai / DeepSpeed-Kernels
  ☆74Updated 3 months ago
• IST-DASLab / QIGen
  Repository for CPU Kernel Generation for LLM Inference
  ☆26Updated 2 years ago
• tridao / flash-attention-wheels
  ☆51Updated last year
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆62Updated 4 months ago
• argonne-lcf / LLM-Inference-Bench
  LLM-Inference-Bench
  ☆45Updated last month
• Raincleared-Song / sparse_gpu_operator
  GPU operators for sparse tensor operations
  ☆33Updated last year
• vedantroy / gpu_kernels
  ☆26Updated last year
• tile-ai / AttentionEngine
  ☆49Updated last month
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆80Updated 10 months ago
• chu-tianxiang / QuIP-for-all
  QuIP quantization
  ☆54Updated last year
• stanford-futuredata / stk
  ☆106Updated 10 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆116Updated 7 months ago
• INT-FlashAttention2024 / INT-FlashAttention
  ☆77Updated 5 months ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆70Updated last year
• Jokeren / triton-samples
  ☆28Updated 6 months ago
• ScalingIntelligence / hydragen
  Hydragen: High-Throughput LLM Inference with Shared Prefixes
  ☆40Updated last year
• fw-ai / llama-cuda-graph-example
  Example of applying CUDA graphs to LLaMA-v2
  ☆12Updated last year
• merrymercy / Awesome-Efficient-LLM
  A curated list for Efficient Large Language Models
  ☆11Updated last year
• SqueezeBits / QUICK
  QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference
  ☆117Updated last year
• IST-DASLab / SparseFinetuning
  Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry
  ☆42Updated last year
• Infini-AI-Lab / MagicDec
  [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding
  ☆121Updated 7 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆79Updated last week
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆47Updated 11 months ago
• Dao-AILab / grouped-latent-attention
  ☆119Updated last month
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆13Updated 11 months ago
• DataStates / datastates-llm
  LLM checkpointing for DeepSpeed/Megatron
  ☆19Updated this week
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆64Updated last year
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆68Updated 4 months ago
• groq / mlagility
  Machine Learning Agility (MLAgility) benchmark and benchmarking tools
  ☆39Updated 2 months ago
• tanyuqian / redco
  NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
  ☆66Updated 7 months ago