A collection of memory efficient attention operators implemented in the Triton language.
-
Updated
Jun 5, 2024 - Python
#
triton-lang
Here are 11 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
Dec 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
🌳️🌐️#️⃣️ The Bliss Browser Triton (ClosedAI) language support module, allowing Triton (ClosedAI) programs to be written in and ran within the browser.
browser web-development gplv3 triton web-browser gpl3 bliss-browser bliss-browser-development bliss-browser-languages bliss-browser-language-support bliss-browser-languagesupport triton-lang triton-language bliss-browser-triton
-
Updated
Dec 18, 2024 - HTML
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."