ailzhang / EfficientPyTorchLinks
Code release for book "Efficient Training in PyTorch"
☆73Updated 3 months ago
Alternatives and similar repositories for EfficientPyTorch
Users that are interested in EfficientPyTorch are comparing it to the libraries listed below
Sorting:
- ⚡️FFPA: Extend FlashAttention-2 with Split-D, achieve ~O(1) SRAM complexity for large headdim, 1.8x~3x↑ vs SDPA.🎉☆192Updated 2 months ago
- flash attention tutorial written in python, triton, cuda, cutlass☆381Updated 2 months ago
- Examples of CUDA implementations by Cutlass CuTe☆206Updated 2 weeks ago
- Puzzles for learning Triton, play it with minimal environment configuration!☆416Updated 7 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆272Updated last year
- Implement Flash Attention using Cute.☆88Updated 7 months ago
- ☆90Updated 3 months ago
- Allow torch tensor memory to be released and resumed later☆80Updated last week
- ☆96Updated 10 months ago
- ☆145Updated 4 months ago
- llm theoretical performance analysis tools and support params, flops, memory and latency analysis.☆98Updated this week
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆87Updated 2 months ago
- ☆172Updated last year
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆108Updated last year
- ☆137Updated last year
- Pipeline Parallelism Emulation and Visualization☆46Updated last month
- ☆149Updated 6 months ago
- ☆84Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆132Updated 6 months ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆183Updated 5 months ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆58Updated 11 months ago
- ☆99Updated 3 months ago
- Tutorials for writing high-performance GPU operators in AI frameworks.☆129Updated last year
- Summary of the Specs of Commonly Used GPUs for Training and Inference of LLM☆53Updated 4 months ago
- ☆139Updated last year
- A Easy-to-understand TensorOp Matmul Tutorial☆365Updated 9 months ago
- High performance Transformer implementation in C++.☆125Updated 6 months ago
- ☆206Updated 7 months ago
- Materials for learning SGLang☆481Updated last week
- A lightweight design for computation-communication overlap.☆146Updated 3 weeks ago