Fetching the paper…

FlashMask: Efficient and Rich Mask Extension of FlashAttention · Around