A collection of memory efficient attention operators implemented in the Triton language.
☆289Jun 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.☆926Updated this week
- Quantized Attention on GPU☆44Nov 22, 2024Updated last year
- ☆109Mar 12, 2026Updated 2 weeks ago
- FlagPerf is an open-source software platform for benchmarking AI chips.☆362Nov 11, 2025Updated 4 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
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Collection of kernels written in Triton language☆185Jan 27, 2026Updated 2 months ago
- flash attention tutorial written in python, triton, cuda, cutlass☆494Jan 20, 2026Updated 2 months ago
- ☆261Jul 11, 2024Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆598Aug 12, 2025Updated 7 months ago
- Cataloging released Triton kernels.☆300Sep 9, 2025Updated 6 months ago
- ☆310Mar 22, 2026Updated last week
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆82Aug 12, 2024Updated last year
- Benchmark tests supporting the TiledCUDA library.☆18Nov 19, 2024Updated last year
- Examples of CUDA implementations by Cutlass CuTe☆271Jul 1, 2025Updated 8 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆46Jun 11, 2025Updated 9 months ago
- A Easy-to-understand TensorOp Matmul Tutorial☆422Mar 5, 2026Updated 3 weeks ago
- Applied AI experiments and examples for PyTorch☆319Aug 22, 2025Updated 7 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.☆98Sep 19, 2025Updated 6 months ago
- Implement Flash Attention using Cute.☆103Dec 17, 2024Updated last year
- Tile primitives for speedy kernels☆3,244Mar 17, 2026Updated last week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆335Updated this week
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆255Feb 13, 2026Updated last month
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- Distributed Compiler based on Triton for Parallel Systems☆1,398Mar 11, 2026Updated 2 weeks ago
- 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…☆3,977Updated this week
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆753Aug 6, 2025Updated 7 months ago
- how to optimize some algorithm in cuda.☆2,887Updated this week
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,692Updated this week
- A Quirky Assortment of CuTe Kernels☆863Mar 22, 2026Updated last week
- Triton-based implementation of Sparse Mixture of Experts.☆270Oct 3, 2025Updated 5 months ago
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆341Feb 23, 2025Updated last year
- FlashInfer: Kernel Library for LLM Serving☆5,231Updated this week
- ☆88Jan 23, 2025Updated last year
- Transformers components but in Triton☆34May 9, 2025Updated 10 months ago
- DeeperGEMM: crazy optimized version☆75May 5, 2025Updated 10 months ago
- Shared Middle-Layer for Triton Compilation☆330Dec 5, 2025Updated 3 months ago
- FlagScale is a large model toolkit based on open-sourced projects.☆497Updated this week