AlibabaPAI / torchaccLinks
PyTorch distributed training acceleration framework
☆49Updated 4 months ago
Alternatives and similar repositories for torchacc
Users that are interested in torchacc are comparing it to the libraries listed below
Sorting:
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆55Updated 10 months ago
- Fast and easy distributed model training examples.☆12Updated 7 months ago
- ☆96Updated 9 months ago
- A prefill & decode disaggregated LLM serving framework with shared GPU memory and fine-grained compute isolation.☆87Updated last month
- A lightweight design for computation-communication overlap.☆143Updated this week
- ☆148Updated 5 months ago
- ☆128Updated 6 months ago
- ☆86Updated 3 months ago
- ☆90Updated 5 months ago
- High performance Transformer implementation in C++.☆125Updated 5 months ago
- ☆103Updated 7 months ago
- nnScaler: Compiling DNN models for Parallel Training☆113Updated this week
- TePDist (TEnsor Program DISTributed) is an HLO-level automatic distributed system for DL models.☆94Updated 2 years ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆144Updated 3 years ago
- Pipeline Parallelism Emulation and Visualization☆43Updated 2 weeks ago
- ☆141Updated 3 months ago
- ☆139Updated last year
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆100Updated last year
- DeepSeek-V3/R1 inference performance simulator☆149Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆127Updated 5 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆79Updated 7 months ago
- A low-latency & high-throughput serving engine for LLMs☆380Updated 3 weeks ago
- ☆77Updated last month
- FlexFlow Serve: Low-Latency, High-Performance LLM Serving☆42Updated last month
- Dynamic Memory Management for Serving LLMs without PagedAttention☆397Updated 3 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆272Updated last year
- ☆117Updated last month
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆38Updated 3 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆80Updated last month
- ☆69Updated this week