FlashMask: Efficient and Rich Mask Extension of FlashAttention

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Wang, Guoxia, Zeng, Jinle, Xiao, Xiyuan, Wu, Siming, Yang, Jiabin, Zheng, Lujing, Chen, Zeyu, Bian, Jiang, Yu, Dianhai, Wang, Haifeng
Format:	Preprint
Published:	2024
Subjects:	Machine Learning
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Internet

https://arxiv.org/abs/2410.01359