A collection of memory efficient attention operators implemented in the Triton language.
-
Updated
Jun 5, 2024 - Python
#
triton-lang
Here are 10 public repositories matching this topic...
Triton implementation of FlashAttention2 that adds Custom Masks.
deep-learning
triton
attention
cuda-kernels
attention-mechanism
triton-lang
flash-attention
flash-attention-2
-
Updated
Aug 14, 2024 - Python
VIT inference in triton because, why not?
-
Updated
May 31, 2024 - Python
Triton implement of bi-directional (non-causal) linear attention
-
Updated
Nov 25, 2024 - Python
🧠️🖥️2️⃣️0️⃣️0️⃣️1️⃣️💾️📜️ The sourceCode:Triton category for AI2001, containing Triton programming language datasets
ai
gplv3
artificial-intelligence
dataset
triton
gpl3
ai2001
ai-2001
ai2001-dataset
ai-2001-dataset
ai2001-development
ai-2001-development
source-code-dataset
triton-lang
sc-dataset
triton-language
triton-sc-dataset
-
Updated
Dec 22, 2023 - R
LAMB go brrr
-
Updated
Apr 11, 2024 - Python
A container of various PyTorch neural network modules written in Triton.
-
Updated
Nov 26, 2024 - Python
Writing TensorRT plugins using Triton and Python
-
Updated
Jul 22, 2024 - Jupyter Notebook
Triton implementation for FISTA (Experimental)
-
Updated
Nov 1, 2024 - Python
Improve this page
Add a description, image, and links to the triton-lang topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the triton-lang topic, visit your repo's landing page and select "manage topics."