ailzhang / EfficientPyTorch
Code release for book "Efficient Training in PyTorch"
☆39Updated 3 months ago
Alternatives and similar repositories for EfficientPyTorch:
Users that are interested in EfficientPyTorch are comparing it to the libraries listed below
- Implement Flash Attention using Cute.☆67Updated last month
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆175Updated this week
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆47Updated this week
- play gemm with tvm☆85Updated last year
- ☆80Updated last year
- Standalone Flash Attention v2 kernel without libtorch dependency☆99Updated 4 months ago
- This is a demo how to write a high performance convolution run on apple silicon☆52Updated 2 years ago
- ☆67Updated last month
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆88Updated 11 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆77Updated 2 months ago
- ☆79Updated 4 months ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated 6 months ago
- 📚[WIP] FFPA: Yet antother Faster Flash Prefill Attention with O(1)⚡️GPU SRAM complexity for headdim > 256, 1.8x~3x↑🎉faster vs SDPA EA.☆73Updated this week
- TileFusion is a highly efficient kernel template library designed to elevate the level of abstraction in CUDA C for processing tiles.☆43Updated this week
- ☆19Updated 4 months ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆53Updated 5 months ago
- GPTQ inference TVM kernel☆38Updated 9 months ago
- A benchmark suited especially for deep learning operators☆42Updated last year
- Examples of CUDA implementations by Cutlass CuTe☆132Updated 2 months ago
- ☆38Updated 7 months ago
- ☆26Updated last week
- Artifacts of EVT ASPLOS'24☆23Updated 10 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆61Updated 2 months ago
- llama INT4 cuda inference with AWQ☆50Updated last week
- ☆70Updated last year
- Puzzles for learning Triton, play it with minimal environment configuration!☆207Updated last month
- Quantized Attention on GPU☆34Updated 2 months ago
- 使用 CUDA C++ 实现的 llama 模型推理框架☆44Updated 2 months ago
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆16Updated 4 months ago