Flex attention integration with block mask cache - #1210

Flex attention integration with block mask cache#1210

djsaunde wants to merge 15 commits into

mainfacebookresearch/fairseq2:mainfrom

flex-attn-block-mask-cachefacebookresearch/fairseq2:flex-attn-block-mask-cacheCopy head branch name to clipboard