Tri Dao, Daniel Y. Fu, Stefano Ermon, Atri Rudra, Christopher RĂ©. FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness. In Sanmi Koyejo, S. Mohamed, A. Agarwal, Danielle Belgrave, K. Cho, A. Oh, editors, Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, NeurIPS 2022, New Orleans, LA, USA, November 28 - December 9, 2022. 2022. [doi]
Abstract is missing.