Fetching the paper…

Flex Attention: A Programming Model for Generating Optimized Attention Kernels · Around