🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.
☆251Feb 13, 2026Updated 2 weeks ago
Alternatives and similar repositories for ffpa-attn
Users that are interested in ffpa-attn are comparing it to the libraries listed below
Sorting:
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆147May 10, 2025Updated 9 months ago
- ☆65Apr 26, 2025Updated 10 months ago
- Quantized Attention on GPU☆44Nov 22, 2024Updated last year
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆106Jun 28, 2025Updated 8 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆97Sep 19, 2025Updated 5 months ago
- Distributed Compiler based on Triton for Parallel Systems☆1,371Feb 13, 2026Updated 2 weeks ago
- flash attention tutorial written in python, triton, cuda, cutlass☆488Jan 20, 2026Updated last month
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆193Jan 28, 2025Updated last year
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,261Aug 28, 2025Updated 6 months ago
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆526Sep 8, 2024Updated last year
- ☆261Jul 11, 2024Updated last year
- 📚LeetCUDA: Modern CUDA Learn Notes with PyTorch for Beginners🐑, 200+ CUDA Kernels, Tensor Cores, HGEMM, FA-2 MMA.🎉☆9,755Updated this week
- A Easy-to-understand TensorOp Matmul Tutorial☆410Feb 11, 2026Updated 2 weeks ago
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆21Feb 9, 2026Updated 3 weeks ago
- ☆20Sep 28, 2024Updated last year
- A lightweight design for computation-communication overlap.☆223Jan 20, 2026Updated last month
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆79Aug 12, 2024Updated last year
- how to optimize some algorithm in cuda.☆2,825Feb 15, 2026Updated 2 weeks ago
- FlagGems is an operator library for large language models implemented in the Triton Language.☆904Updated this week
- FlashInfer: Kernel Library for LLM Serving☆5,057Updated this week
- Fast low-bit matmul kernels in Triton☆433Feb 1, 2026Updated last month
- ☆34Feb 3, 2025Updated last year
- ☆62Feb 15, 2026Updated 2 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆288Jun 5, 2024Updated last year
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆277Jul 16, 2025Updated 7 months ago
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆20Aug 3, 2025Updated 7 months ago
- ☆52May 19, 2025Updated 9 months ago
- Efficient LLM Inference over Long Sequences☆393Jun 25, 2025Updated 8 months ago
- A throughput-oriented high-performance serving framework for LLMs☆946Oct 29, 2025Updated 4 months ago
- Implement Flash Attention using Cute.☆101Dec 17, 2024Updated last year
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆5,284Updated this week
- DeeperGEMM: crazy optimized version☆74May 5, 2025Updated 9 months ago
- An experimental communicating attention kernel based on DeepEP.☆35Jul 29, 2025Updated 7 months ago
- Applied AI experiments and examples for PyTorch☆319Aug 22, 2025Updated 6 months ago
- Writing a CUDA software ray tracing renderer with Analysis-Driven Optimization from scratch: a python-importable, distributed parallel re…☆37Oct 5, 2025Updated 4 months ago
- kernels, of the mega variety☆679Jan 29, 2026Updated last month
- ☆118May 19, 2025Updated 9 months ago
- Benchmark tests supporting the TiledCUDA library.☆18Nov 19, 2024Updated last year
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆161Oct 13, 2025Updated 4 months ago