DeepLink-org / DIOPI
☆68Updated 2 months ago
Alternatives and similar repositories for DIOPI:
Users that are interested in DIOPI are comparing it to the libraries listed below
- ☆65Updated 3 months ago
- A benchmark suited especially for deep learning operators☆42Updated 2 years ago
- ☆12Updated 3 weeks ago
- Development repository for the Triton-Linalg conversion☆173Updated 2 weeks ago
- ☆142Updated last month
- ☆140Updated 9 months ago
- ☆127Updated last month
- ☆98Updated 2 months ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆296Updated 3 months ago
- Examples of CUDA implementations by Cutlass CuTe☆138Updated 2 weeks ago
- FlagGems is an operator library for large language models implemented in Triton Language.☆421Updated this week
- PyTorch distributed training acceleration framework☆39Updated last week
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆51Updated 6 months ago
- A Easy-to-understand TensorOp Matmul Tutorial☆316Updated 5 months ago
- ☆57Updated 3 months ago
- ☆110Updated 11 months ago
- Yinghan's Code Sample☆305Updated 2 years ago
- ☆129Updated last month
- A collection of memory efficient attention operators implemented in the Triton language.☆240Updated 8 months ago
- ☆35Updated 4 months ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆142Updated 2 years ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆35Updated 5 months ago
- learning how CUDA works☆201Updated 6 months ago
- A tutorial for CUDA&PyTorch☆126Updated last month
- Fast and easy distributed model training examples.☆11Updated 2 months ago
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆345Updated 5 months ago
- flash attention tutorial written in python, triton, cuda, cutlass☆260Updated last month
- Dynamic Memory Management for Serving LLMs without PagedAttention☆290Updated this week
- A simple high performance CUDA GEMM implementation.☆347Updated last year
- ☆81Updated 5 months ago