Pranshu Malviya, Gonçalo Mordido, Aristide Baratin, Reza Babanezhad Harikandeh, Jerry Huang, Simon Lacoste-Julien, Razvan Pascanu, Sarath Chandar. Promoting Exploration in Memory-Augmented Adam using Critical Momenta. Trans. Mach. Learn. Res., 2024, 2024. [doi]