AlibabaPAI / torchacc
PyTorch distributed training acceleration framework
☆32Updated 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
- ☆79Updated 2 months ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆217Updated last week
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆200Updated 3 weeks ago
- 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
- High performance Transformer implementation in C++.☆78Updated last month
- Dynamic Memory Management for Serving LLMs without PagedAttention☆222Updated this week
- ☆136Updated this week
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆75Updated this week
- Paella: Low-latency Model Serving with Virtualized GPU Scheduling☆57Updated 6 months ago
- ☆123Updated this week
- Artifact of OSDI '24 paper, ”Llumnix: Dynamic Scheduling for Large Language Model Serving“☆56Updated 5 months ago
- A low-latency & high-throughput serving engine for LLMs☆231Updated last month
- ☆140Updated 6 months ago
- Efficient and easy multi-instance LLM serving☆192Updated this week
- TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.☆148Updated this week
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆46Updated 3 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆114Updated 2 years ago
- ☆33Updated 2 months ago
- nnScaler: Compiling DNN models for Parallel Training☆62Updated 2 weeks ago
- ☆162Updated 3 months ago
- ☆78Updated 8 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆26Updated 2 months ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆246Updated this week
- ☆72Updated last year
- Since the emergence of chatGPT in 2022, the acceleration of Large Language Model has become increasingly important. Here is a list of pap…☆166Updated this week
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆74Updated 7 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆85Updated 8 months ago
- ☆47Updated 2 weeks ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆142Updated 2 years ago