@misc{indiciaea37c9293f9fc, title = {Flex Attention: A Programming Model for Generating Optimized Attention Kernels}, author = {Juechu Dong and Boyuan Feng and Driss Guessous and Yanbo Liang and Horace He}, year = {2024}, url = {https://arxiv.org/abs/2412.05496}, note = {Source identifier: 2412.05496} }