FlashMask: Efficient and Rich Mask Extension of FlashAttention

Open in new window