How to Capture Higher-order Correlations? Generalizing Matrix Softmax Attention to Kronecker Computation - Alman, Josh; Song, Zhao (International Conference on Learning Representations, 2024)