ailzhang / EfficientPyTorchLinks
Code release for book "Efficient Training in PyTorch"
☆125Updated 9 months ago
Alternatives and similar repositories for EfficientPyTorch
Users that are interested in EfficientPyTorch are comparing it to the libraries listed below
Sorting:
- 分享AI Infra知识&代码练习:PyTorch/vLLM/SGLang框架入门⚡️、性能加速🚀、大模型基础🧠、AI软硬件🔧等☆215Updated this week
- Puzzles for learning Triton, play it with minimal environment configuration!☆613Updated last month
- Examples of CUDA implementations by Cutlass CuTe☆270Updated 7 months ago
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆250Updated this week
- Flash Attention from Scratch on CUDA Ampere☆129Updated 5 months ago
- llm theoretical performance analysis tools and support params, flops, memory and latency analysis.☆115Updated 6 months ago
- 注释的nano_vllm仓库,并且完成了MiniCPM4的适配以及注册新模型的功能☆155Updated 5 months ago
- Codes & examples for "CUDA - From Correctness to Performance"☆121Updated last year
- ☆117Updated 4 months ago
- ☆177Updated 2 years ago
- flash attention tutorial written in python, triton, cuda, cutlass☆484Updated 2 weeks ago
- Summary of the Specs of Commonly Used GPUs for Training and Inference of LLM☆73Updated 5 months ago
- UltraScale Playbook 中文版☆125Updated 10 months ago
- Train speculative decoding models effortlessly and port them smoothly to SGLang serving.☆676Updated this week
- Pipeline Parallelism Emulation and Visualization☆77Updated 3 weeks ago
- ☆155Updated 11 months ago
- ☆152Updated 7 months ago
- ☆96Updated 10 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆287Updated last year
- Allow torch tensor memory to be released and resumed later☆216Updated 3 weeks ago
- ☆105Updated last year
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆148Updated 8 months ago
- High Performance LLM Inference Operator Library☆695Updated this week
- 🌈 Solutions of LeetGPU☆70Updated last week
- learning how CUDA works☆373Updated 11 months ago
- Tutorials for writing high-performance GPU operators in AI frameworks.☆136Updated 2 years ago
- Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend☆107Updated this week
- High performance Transformer implementation in C++.☆150Updated last year
- Implement Flash Attention using Cute.☆100Updated last year
- ☆113Updated 8 months ago