alexzhang13 / flashattention2-custom-mask

Triton implementation of FlashAttention2 that adds Custom Masks.

Date Created 2024-07-20 (5 months ago)

Commits 18 (last one 4 months ago)

Stargazers 86 (2 this week)

Watchers 5 (0 this week)

Forks 8

License apache-2.0

Ranking

RepositoryStats indexes 595,856 repositories, of these alexzhang13/flashattention2-custom-mask is ranked #325,826 (45th percentile) for total stargazers, and #335,688 for total watchers. Github reports the primary language for this repository as Python, for repositories using this language it is ranked #61,043/119,431.

alexzhang13/flashattention2-custom-mask is also tagged with popular topics, for these it's ranked: deep-learning (#5,455/8512)

Other Information

alexzhang13/flashattention2-custom-mask has Github issues enabled, there are 9 open issues and 3 closed issues.

All Topics

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

Star History

Github stargazers over time

Watcher History

Github watchers over time, collection started in '23

Recent Commit History

18 commits on the default branch (main) since jan '22

Yearly Commits

Commits to the default branch (main) per year

Issue History

Languages

The only known language in this repository is Python

updated: 2024-12-17 @ 08:55pm, id: 831270187 / R_kgDOMYwtKw