AlibabaPAI / FlashModelsLinks
Fast and easy distributed model training examples.
☆12Updated 10 months ago
Alternatives and similar repositories for FlashModels
Users that are interested in FlashModels are comparing it to the libraries listed below
Sorting:
- PyTorch distributed training acceleration framework☆52Updated 2 months ago
- TePDist (TEnsor Program DISTributed) is an HLO-level automatic distributed system for DL models.☆97Updated 2 years ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆65Updated last year
- Zero Bubble Pipeline Parallelism☆432Updated 5 months ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆146Updated 3 years ago
- A Easy-to-understand TensorOp Matmul Tutorial☆381Updated last week
- ☆148Updated 5 months ago
- ☆150Updated 9 months ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆426Updated 4 months ago
- ☆136Updated 10 months ago
- Perplexity GPU Kernels☆488Updated last month
- Examples of CUDA implementations by Cutlass CuTe☆241Updated 3 months ago
- ☆240Updated last year
- ☆124Updated 9 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆281Updated last year
- A lightweight design for computation-communication overlap.☆181Updated last week
- Yinghan's Code Sample☆352Updated 3 years ago
- ☆100Updated last year
- ☆107Updated 5 months ago
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆265Updated 2 months ago
- Development repository for the Triton-Linalg conversion☆202Updated 8 months ago
- ☆139Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆154Updated last week
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆421Updated last week
- GLake: optimizing GPU memory management and IO transmission.☆481Updated 6 months ago
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆479Updated last year
- Ongoing research training transformer models at scale☆18Updated last week
- nnScaler: Compiling DNN models for Parallel Training☆118Updated 3 weeks ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆172Updated 3 weeks ago
- Pipeline Parallelism Emulation and Visualization☆67Updated 4 months ago