Flashlight: PyTorch Compiler Extensions to Accelerate Attention Variants
Published in Machine Learning and Systems (MLSys), 2026
Recommended citation: Bozhi You, Irene Wang, Zelal Su Mustafaoglu, Abhinav Jangda, Angélica Moreira, Roshan Dathathri, Divya Mahajan, Keshav Pingali, “Flashlight: PyTorch Compiler Extensions to Accelerate Attention Variants,” Proceedings of the Ninth Conference on Machine Learning and Systems (MLSys), May 2026. https://openreview.net/forum?id=lboOMA8XWr
