Flashlight: PyTorch Compiler Extensions to Accelerate Attention Variants. - Bozhi You; Irene Wang; Zelal Su Mustafaoglu; Abhinav Jangda; Angélica Moreira; Roshan Dathathri; Divya Mahajan; Keshav Pingali (CoRR, 2025)