Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep Learning - Kossen, Jannik; Band, Neil; Lyle, Clare; Gomez, Aidan N; Rainforth, Thomas; Gal, Yarin (Advances in Neural Information Processing Systems, 2021)