Search Results - RepositoryStats

18

233

other

6

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

attention triton-lang

Created 2023-10-07

47 commits to main branch, last one 8 months ago

8

91

apache-2.0

4

Triton implementation of FlashAttention2 that adds Custom Masks.

triton attention triton-lang cuda-kernels deep-learning flash-attention flash-attention-2 attention-mechanism

Created 2024-07-20

18 commits to main branch, last one 5 months ago

1

39

mit

2

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

triton-lang computer-vision machine-learning-systems

Created 2024-11-20

31 commits to main branch, last one 21 days ago