hpcaitech / ColossalAI-Benchmark
Performance benchmarking with ColossalAI
☆39Updated 2 years ago
Alternatives and similar repositories for ColossalAI-Benchmark:
Users that are interested in ColossalAI-Benchmark are comparing it to the libraries listed below
- A Python library transfers PyTorch tensors between CPU and NVMe☆102Updated 2 months ago
- Scalable PaLM implementation of PyTorch☆192Updated 2 years ago
- ☆26Updated 3 years ago
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆90Updated 2 years ago
- ☆70Updated 3 years ago
- ☆72Updated 2 years ago
- ☆97Updated 5 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆77Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆86Updated 3 weeks ago
- PyTorch bindings for CUTLASS grouped GEMM.☆61Updated 3 months ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆204Updated 5 months ago
- Zero Bubble Pipeline Parallelism☆317Updated 2 months ago
- A high-performance distributed deep learning system targeting large-scale and automated distributed training. If you have any interests, …☆106Updated last year
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆37Updated 2 years ago
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 7 months ago
- ☆41Updated 2 years ago
- Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelines.☆52Updated last year
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆55Updated 3 years ago
- ☆79Updated 2 months ago
- ATC23 AE☆44Updated last year
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆47Updated 6 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆195Updated last year
- Examples of training models with hybrid parallelism using ColossalAI☆337Updated last year
- nnScaler: Compiling DNN models for Parallel Training☆87Updated 3 weeks ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆142Updated 2 years ago
- PyTorch implementation of paper "Response Length Perception and Sequence Scheduling: An LLM-Empowered LLM Inference Pipeline".☆81Updated last year
- ☆79Updated 4 months ago
- ☆140Updated 9 months ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆135Updated 8 months ago
- ☆134Updated 6 months ago