Transformer Memory as a Differentiable Search Index - Tay, Yi; Tran, Vinh; Dehghani, Mostafa; Ni, Jianmo; Bahri, Dara; Mehta, Harsh; Qin, Zhen; Hui, Kai; Zhao, Zhe; Gupta, Jai; Schuster, Tal; Cohen, William W; et al. (Advances in Neural Information Processing Systems, 2022)