Loki: Low-rank Keys for Efficient Sparse Attention - Singhania, Prajwal; Singh, Siddharth; He, Shwai; Feizi, Soheil; Bhatele, Abhinav (Advances in Neural Information Processing Systems, 2024)