hpcaitech / ColossalAI-Benchmark
Performance benchmarking with ColossalAI
☆38Updated 2 years ago
Related projects: ⓘ
- A Python library transfers PyTorch tensors between CPU and NVMe☆92Updated last year
- Scalable PaLM implementation of PyTorch☆191Updated last year
- ☆25Updated 3 years ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆183Updated last month
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆70Updated last month
- ☆62Updated 3 years ago
- ☆65Updated 2 years ago
- Zero Bubble Pipeline Parallelism☆254Updated 2 weeks ago
- PyTorch bindings for CUTLASS grouped GEMM.☆57Updated 2 months ago
- ☆83Updated 3 weeks ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆161Updated 2 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆166Updated 11 months ago
- PyTorch library for cost-effective, fast and easy serving of MoE models.☆90Updated last month
- ATC23 AE☆42Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆309Updated this week
- ☆67Updated last week
- ☆68Updated 9 months ago
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆89Updated last year
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆41Updated last month
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆173Updated 3 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆205Updated 3 months ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆138Updated 2 years ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆184Updated this week
- Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelines.☆41Updated 9 months ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆186Updated last month
- ☆127Updated last month
- A high-performance distributed deep learning system targeting large-scale and automated distributed training. If you have any interests, …☆101Updated 9 months ago
- Latency and Memory Analysis of Transformer Models for Training and Inference☆338Updated 3 months ago
- PyTorch implementation of paper "Response Length Perception and Sequence Scheduling: An LLM-Empowered LLM Inference Pipeline".☆70Updated last year
- ☆39Updated last year