Triton implementation of FlashAttention2 that adds Custom Masks. - View it on GitHub