DeepLink-org / DIOPI
☆68Updated 2 weeks ago
Related projects: ⓘ
- ☆56Updated last week
- A benchmark suited especially for deep learning operators☆40Updated last year
- ☆133Updated 2 months ago
- ☆11Updated last week
- FlagGems is an operator library for large language models implemented in Triton Language.☆246Updated last week
- Development repository for the Triton-Linalg conversion☆137Updated last month
- Dynamic Memory Management for Serving LLMs without PagedAttention☆186Updated last month
- A fast communication-overlapping library for tensor parallelism on GPUs.☆184Updated this week
- ☆140Updated 4 months ago
- A Easy-to-understand TensorOp Matmul Tutorial☆265Updated this week
- A collection of memory efficient attention operators implemented in the Triton language.☆205Updated 3 months ago
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆266Updated last week
- ☆70Updated 6 months ago
- flash attention tutorial written in python, triton, cuda, cutlass☆159Updated 3 months ago
- ☆95Updated 2 years ago
- ☆123Updated 3 months ago
- ☆56Updated last week
- Disaggregated serving system for Large Language Models (LLMs).☆278Updated last month
- learning how CUDA works☆150Updated last month
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆41Updated last month
- Yinghan's Code Sample☆272Updated 2 years ago
- ☆90Updated 6 months ago
- GLake: optimizing GPU memory management and IO transmission.☆351Updated last month
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆20Updated last week
- A baseline repository of Auto-Parallelism in Training Neural Networks☆138Updated 2 years ago
- A model compilation solution for various hardware☆357Updated last week
- A simple high performance CUDA GEMM implementation.☆319Updated 8 months ago
- Zero Bubble Pipeline Parallelism☆254Updated 2 weeks ago
- PyTorch distributed training acceleration framework☆16Updated this week
- High performance Transformer implementation in C++.☆67Updated last week