Alternatives and similar repositories for Megatron-LM

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

• ROCm / TransformerEngine
  ☆34Updated this week
• ROCm / MAD
  ☆18Updated this week
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆64Updated this week
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆76Updated this week
• ROCm / pytorch-micro-benchmarking
  ☆20Updated last month
• intel / intel-extension-for-deepspeed
  Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…
  ☆61Updated 2 months ago
• Azure / msccl
  Microsoft Collective Communication Library
  ☆65Updated 5 months ago
• argonne-lcf / LLM-Inference-Bench
  LLM-Inference-Bench
  ☆40Updated 4 months ago
• yifuwang / symm-mem-recipes
  ☆76Updated 4 months ago
• microsoft / nnscaler
  nnScaler: Compiling DNN models for Parallel Training
  ☆110Updated 2 weeks ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆89Updated 2 weeks ago
• merthidayetoglu / HiCCL
  A hierarchical collective communications library with portable optimizations
  ☆35Updated 5 months ago
• codeplaysoftware / cutlass-sycl
  A CUTLASS implementation using SYCL
  ☆22Updated this week
• ROCm / rocmProfileData
  ☆24Updated last week
• HabanaAI / vllm-fork
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆70Updated this week
• ColfaxResearch / cfx-article-src
  ☆109Updated last week
• ColfaxResearch / cutlass-kernels
  ☆202Updated 10 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆109Updated 10 months ago
• Azure / msccl-executor-nccl
  ☆36Updated 5 months ago
• ROCm / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆22Updated this week
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆248Updated 6 months ago
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆121Updated 4 months ago
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆174Updated this week
• gty111 / GEMM_MMA
  Optimize GEMM with tensorcore step by step
  ☆26Updated last year
• triton-lang / kernels
  ☆79Updated 6 months ago
• abhibambhaniya / GenZ-LLM-Analyzer
  LLM Inference analyzer for different hardware platforms
  ☆66Updated 2 weeks ago
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆113Updated last week
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆267Updated this week
• shixun404 / Fault-Tolerant-SGEMM-on-NVIDIA-GPUs
  Anatomy of High-Performance GEMM with Online Fault Tolerance on GPUs
  ☆12Updated last month
• shenh10 / DeepSeek_Simulator
  ☆65Updated last month