AlibabaPAI / torchacc
PyTorch distributed training acceleration framework
☆34Updated this week
Related projects ⓘ
Alternatives and complementary repositories for torchacc
- TePDist (TEnsor Program DISTributed) is an HLO-level automatic distributed system for DL models.☆90Updated last year
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆203Updated last week
- A low-latency & high-throughput serving engine for LLMs☆245Updated 2 months ago
- ☆79Updated 2 months ago
- High performance Transformer implementation in C++.☆82Updated 2 months ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆224Updated 3 weeks ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆238Updated last week
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆133Updated 2 months ago
- Artifact of OSDI '24 paper, ”Llumnix: Dynamic Scheduling for Large Language Model Serving“☆57Updated 5 months ago
- ☆140Updated 6 months ago
- nnScaler: Compiling DNN models for Parallel Training☆74Updated 3 weeks ago
- Paella: Low-latency Model Serving with Virtualized GPU Scheduling☆57Updated 6 months ago
- ☆123Updated 2 weeks ago
- ☆138Updated 2 weeks ago
- Disaggregated serving system for Large Language Models (LLMs).☆359Updated 3 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆75Updated this week
- A baseline repository of Auto-Parallelism in Training Neural Networks☆142Updated 2 years ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆45Updated 3 months ago
- ☆33Updated 2 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆219Updated 5 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆114Updated 2 years ago
- ☆64Updated 3 months ago
- Materials for learning SGLang☆96Updated this week
- ☆73Updated last year
- ☆52Updated last week
- Efficient and easy multi-instance LLM serving☆213Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆68Updated 4 months ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆75Updated 8 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆29Updated 2 months ago
- ☆65Updated 3 years ago