Faster Neighborhood Attention: Reducing the O(n^2) Cost of Self Attention at the Threadblock Level - Hassani, Ali; Hwu, Wen-mei; Shi, Humphrey (Advances in Neural Information Processing Systems, 2024)