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☆120Updated 10 months ago
- ☆112Updated last year
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆215Updated last year
- ☆28Updated 4 years ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆66Updated 6 months ago
- ☆30Updated 2 years ago
- PyTorch bindings for CUTLASS grouped GEMM.☆154Updated last week
- ATC23 AE☆47Updated 2 years ago
- ☆87Updated 3 years ago
- ☆75Updated 4 years ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆80Updated 11 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆77Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆124Updated 4 months ago
- [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
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆221Updated 2 years ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆41Updated 2 years ago
- nnScaler: Compiling DNN models for Parallel Training☆118Updated 3 weeks ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆265Updated 3 months ago
- Scalable PaLM implementation of PyTorch☆188Updated 2 years ago
- Pipeline Parallelism Emulation and Visualization☆67Updated 4 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆281Updated last year
- Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.☆108Updated last week
- ☆121Updated last year
- Zero Bubble Pipeline Parallelism☆432Updated 5 months ago
- ☆72Updated 6 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆215Updated this week
- pytorch-profiler☆51Updated 2 years ago
- ☆158Updated 2 years ago
- ☆41Updated last year
- ☆42Updated 2 years ago