Mastering Memory Tasks with World Models - Samsami, Mohammad Reza; Zholus, Artem; Rajendran, Janarthanan; Chandar, Sarath (International Conference on Learning Representations, 2024)