matlok 's Collections
LMM

Papers - Attention - Flex Attention

https://pytorch.org/blog/flexattention/