triton-lang

Puzzles for learning Triton, play it with minimal environment configuration!

A collection of memory efficient attention operators implemented in the Triton language.

Triton implementation of FlashAttention2 that adds Custom Masks.

Triton implement of bi-directional (non-causal) linear attention

A Triton JIT runtime and ffi provider in C++

VIT inference in triton because, why not?

A "standard library" of Triton kernels.

Collection of useful kernels

[ICML'26] Beyond Test-Time Memory: State-Space Optimal Control for LLM Reasoning

Educational resource demonstrating common GPU programming pitfalls and solutions using Triton kernels.

Experimental Rust DSL for writing GPU kernels that compile through the Triton compiler — no Python required.

LAMB go brrr

A from-scratch nano-Triton and nano-Helion: the modern GPU-kernel DSL stack, rebuilt in ~4,000 lines

🧠️🖥️2️⃣️0️⃣️0️⃣️1️⃣️💾️📜️ The sourceCode:Triton category for AI2001, containing Triton programming language datasets

FlashAttention implementations using CUDA and Triton

collection of high-performance CUDA implementations, ranging from naive to highly optimized versions.

A container of various PyTorch neural network modules written in Triton.

🌳️🌐️#️⃣️ The Bliss Browser Triton (ClosedAI) language support module, allowing Triton (ClosedAI) programs to be written in and ran within the browser.

Writing TensorRT plugins using Triton and Python

Fast Triton kernels for multi-scale deformable attention (MSDA) — the core operator behind Deformable DETR, DINO, and Mask2Former — plus a from-scratch course and hands-on labs