Escaping the Gravitational Pull of Softmax - Mei, Jincheng; Xiao, Chenjun; Dai, Bo; Li, Lihong; Szepesvari, Csaba; Schuurmans, Dale (Advances in Neural Information Processing Systems, 2020)