FlashMask: Efficient and Rich Mask Extension of FlashAttention

Open Webpage

Guoxia Wang, Jinle Zeng, Xiyuan Xiao, Siming Wu, Jiabin Yang, Lujing Zheng, Zeyu Chen, Jiang Bian, Dianhai Yu, Haifeng Wang

Published: 01 Jan 2025, Last Modified: 08 Oct 2025ICLR 2025EveryoneRevisionsBibTeXCC BY-SA 4.0