alexzhang13/flashattention2-custom-mask

Triton implementation of FlashAttention2 that adds Custom Masks.

1 best points · 1 comments · 1 Hacker News post · 180 GitHub stars · View on GitHub

Hacker News discussions

• FlashAttention2 with Custom Masking (1 points, 1 comments, by nopon00, 2024-07-21)