FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness - Dao, Tri; Fu, Dan; Ermon, Stefano; Rudra, Atri; Ré, Christopher (Advances in Neural Information Processing Systems, 2022)