Alternatives and similar repositories for ColossalAI-Benchmark

Users that are interested in ColossalAI-Benchmark are comparing it to the libraries listed below

• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆116Updated 7 months ago
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆79Updated 7 months ago
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆212Updated 10 months ago
• xuqifan897 / Optimus
  ☆27Updated 4 years ago
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆132Updated 6 months ago
• stanford-futuredata / stk
  ☆106Updated 10 months ago
• thu-pacman / FasterMoE
  ☆86Updated 3 years ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆101Updated last month
• ParCIS / Chimera
  Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.
  ☆67Updated 3 months ago
• Youhe-Jiang / IJCAI2023-OptimalShardedDataParallel
  [IJCAI2023] An automated parallel training system that combines the advantages from both data and model parallelism. If you have any inte…
  ☆51Updated 2 years ago
• zhuohan123 / terapipe
  ☆75Updated 4 years ago
• sail-sg / zero-bubble-pipeline-parallelism
  Zero Bubble Pipeline Parallelism
  ☆406Updated 2 months ago
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆40Updated 2 years ago
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆272Updated last year
• thu-pacman / SmartMoE-AE
  ATC23 AE
  ☆45Updated 2 years ago
• qhliu26 / Dive-into-Big-Model-Training
  📑 Dive into Big Model Training
  ☆114Updated 2 years ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆70Updated last year
• Hsword / Hetu
  A high-performance distributed deep learning system targeting large-scale and automated distributed training. If you have any interests, …
  ☆115Updated last year
• microsoft / nnscaler
  nnScaler: Compiling DNN models for Parallel Training
  ☆113Updated last week
• hpcaitech / PaLM-colossalai
  Scalable PaLM implementation of PyTorch
  ☆190Updated 2 years ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆255Updated 8 months ago
• HPDL-Group / Merak
  ☆80Updated 2 months ago
• deepspeedai / DeepSpeed-Kernels
  ☆74Updated 3 months ago
• exists-forall / striped_attention
  ☆39Updated last year
• LiuXiaoxuanPKU / GACT-ICML
  ☆42Updated 2 years ago
• AlibabaResearch / flash-llm
  Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity
  ☆216Updated last year
• yanring / Megatron-MoE-ModelZoo
  Best practices for testing advanced Mixtral, DeepSeek, and Qwen series MoE models using Megatron Core MoE.
  ☆32Updated this week
• AlibabaPAI / FLASHNN
  ☆96Updated 10 months ago
• madsys-dev / deepseekv2-profile
  ☆145Updated 4 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆184Updated this week