hpcaitech / ColossalAI-BenchmarkLinks
Performance benchmarking with ColossalAI
☆39Updated 3 years ago
Alternatives and similar repositories for ColossalAI-Benchmark
Users that are interested in ColossalAI-Benchmark are comparing it to the libraries listed below
Sorting:
- A Python library transfers PyTorch tensors between CPU and NVMe☆121Updated 10 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆216Updated last year
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆79Updated 10 months ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆66Updated 6 months ago
- ☆112Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆151Updated 3 weeks ago
- ☆28Updated 4 years ago
- PyTorch bindings for CUTLASS grouped GEMM.☆121Updated 3 months ago
- 📑 Dive into Big Model Training☆114Updated 2 years ago
- A collection of memory efficient attention operators implemented in the Triton language.☆279Updated last year
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆40Updated 2 years ago
- Scalable PaLM implementation of PyTorch☆190Updated 2 years ago
- ☆87Updated 3 years ago
- ☆75Updated 4 years ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆265Updated 2 months ago
- ATC23 AE☆47Updated 2 years ago
- Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.☆93Updated last week
- [IJCAI2023] An automated parallel training system that combines the advantages from both data and model parallelism. If you have any inte…☆52Updated 2 years ago
- Odysseus: Playground of LLM Sequence Parallelism☆77Updated last year
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆91Updated 2 years ago
- nnScaler: Compiling DNN models for Parallel Training☆118Updated 3 weeks ago
- pytorch-profiler☆51Updated 2 years ago
- Utility scripts for PyTorch (e.g. Memory profiler that understands more low-level allocations such as NCCL)☆53Updated 2 weeks ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆222Updated 2 years ago
- ☆81Updated 4 months ago
- Pipeline Parallelism Emulation and Visualization☆66Updated 3 months ago
- ☆74Updated 6 months ago
- Zero Bubble Pipeline Parallelism☆427Updated 4 months ago
- ☆42Updated 2 years ago
- ☆98Updated last year