AlibabaPAI / FlashModels
Fast and easy distributed model training examples.
☆10Updated this week
Related projects ⓘ
Alternatives and complementary repositories for FlashModels
- PyTorch distributed training acceleration framework☆34Updated this week
- A baseline repository of Auto-Parallelism in Training Neural Networks☆141Updated 2 years ago
- ☆73Updated last year
- nnScaler: Compiling DNN models for Parallel Training☆77Updated 3 weeks ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆240Updated last week
- A fast communication-overlapping library for tensor parallelism on GPUs.☆226Updated 3 weeks ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆46Updated 3 months ago
- TePDist (TEnsor Program DISTributed) is an HLO-level automatic distributed system for DL models.☆90Updated last year
- Paella: Low-latency Model Serving with Virtualized GPU Scheduling☆57Updated 6 months ago
- Development repository for the Triton-Linalg conversion☆151Updated last month
- ☆79Updated 8 months ago
- ☆68Updated last week
- ☆31Updated last year
- A benchmark suited especially for deep learning operators☆41Updated last year
- ☆138Updated 2 weeks ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆250Updated this week
- A Easy-to-understand TensorOp Matmul Tutorial☆294Updated 2 months ago
- Efficient and easy multi-instance LLM serving☆219Updated this week
- ☆169Updated 4 months ago
- High performance Transformer implementation in C++.☆83Updated 2 months ago
- Artifact of OSDI '24 paper, ”Llumnix: Dynamic Scheduling for Large Language Model Serving“☆57Updated 5 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆114Updated 2 years ago
- ☆23Updated last year
- TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.☆157Updated this week
- Curated collection of papers in machine learning systems☆167Updated last month
- Yinghan's Code Sample☆289Updated 2 years ago
- ☆66Updated 3 years ago
- GLake: optimizing GPU memory management and IO transmission.☆381Updated 3 months ago
- paper and its code for AI System☆216Updated 2 months 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