alexzhang13 / flashattention2-custom-maskView on GitHub
Triton implementation of FlashAttention2 that adds Custom Masks.
170Aug 14, 2024Updated last year

Alternatives and similar repositories for flashattention2-custom-mask

Users that are interested in flashattention2-custom-mask are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?