Search Results - RepositoryStats

18

253

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 9 months ago

11

102

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 7 months ago

1

44

mit

1

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

triton-lang computer-vision machine-learning-systems

Created 2024-11-20

32 commits to main branch, last one about a month ago

2

26

unknown

0

VIT inference in triton because, why not?

gpu cuda triton-lang deep-learning llm-inference

Created 2024-04-15

34 commits to main branch, last one 9 months ago