A collection of memory efficient attention operators implemented in the Triton language.
☆291Jun 5, 2024Updated last year
Alternatives and similar repositories for FlagAttention
Users that are interested in FlagAttention are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- FlagGems is an operator library for large language models implemented in the Triton Language.☆1,001Updated this week
- Quantized Attention on GPU☆44Nov 22, 2024Updated last year
- ☆113Mar 12, 2026Updated 2 months ago
- 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
- Collection of kernels written in Triton language☆195Jan 27, 2026Updated 4 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆599May 13, 2026Updated 2 weeks ago
- flash attention tutorial written in python, triton, cuda, cutlass☆517Jan 20, 2026Updated 4 months ago
- ☆267Jul 11, 2024Updated last year
- Cataloging released Triton kernels.☆303Sep 9, 2025Updated 8 months ago
- ☆329Updated this week
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆81Aug 12, 2024Updated last year
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- Examples of CUDA implementations by Cutlass CuTe☆274Jul 1, 2025Updated 10 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆47Jun 11, 2025Updated 11 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A Easy-to-understand TensorOp Matmul Tutorial☆438Mar 5, 2026Updated 2 months ago
- Applied AI experiments and examples for PyTorch☆322Aug 22, 2025Updated 9 months ago
- Experiment of using Tangent to autodiff triton☆82Jan 22, 2024Updated 2 years ago
- Framework to reduce autotune overhead to zero for well known deployments.☆100Sep 19, 2025Updated 8 months ago
- Implement Flash Attention using Cute.☆108Dec 17, 2024Updated last year
- Tile primitives for speedy kernels☆3,377May 22, 2026Updated last week
- Distributed Compiler based on Triton for Parallel Systems☆1,440Apr 22, 2026Updated last month
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆355Updated this week
- 🤖FFPA: Extends FlashAttention-2 via Split-D for large headdims, 1.5x~3×↑🎉 vs SDPA, up to 430T🎉 on H200.☆299May 20, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Triton Implementation of HyperAttention Algorithm☆48Dec 11, 2023Updated 2 years ago
- LightLLM is a Python-based LLM (Large Language Model) inference and serving framework, notable for its lightweight design, easy scalabili…☆4,073May 22, 2026Updated last week
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆764Aug 6, 2025Updated 9 months ago
- A Quirky Assortment of CuTe Kernels☆985May 20, 2026Updated last week
- how to optimize some algorithm in cuda.☆2,998May 20, 2026Updated last week
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- 🚀 Efficient implementations for emerging model architectures☆5,139Updated this week
- Triton-based implementation of Sparse Mixture of Experts.☆276Oct 3, 2025Updated 7 months ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆344Feb 23, 2025Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆89Jan 23, 2025Updated last year
- Transformers components but in Triton☆34May 9, 2025Updated last year
- FlashInfer: Kernel Library for LLM Serving☆5,666Updated this week
- Shared Middle-Layer for Triton Compilation☆333Dec 5, 2025Updated 5 months ago
- DeeperGEMM: crazy optimized version☆86May 5, 2025Updated last year
- extensible collectives library in triton☆98Mar 31, 2025Updated last year
- FlagScale is a large model toolkit based on open-sourced projects.☆517Updated this week