Measuring Sample Efficiency and Generalization in Reinforcement Learning Benchmarks: NeurIPS 2020 Procgen Benchmark. - Sharada P. Mohanty; Jyotish Poonganam; Adrien Gaidon; Andrey Kolobov; Blake Wulfe; Dipam Chakraborty; Grazvydas Semetulskis; João Schapke; Jonas Kubilius; Jurgis Pasukonis; Linas Klimas; Matthew J. Hausknecht; et al. (NeurIPS (Competition and Demos), 2020)

Evaluating Long-Term Memory in 3D Mazes. - Jurgis Pasukonis; Timothy P. Lillicrap; Danijar Hafner (ICLR, 2023)

Mastering Diverse Domains through World Models. - Danijar Hafner; Jurgis Pasukonis; Jimmy Ba; Timothy P. Lillicrap (CoRR, 2023)

GATS: Gather-Attend-Scatter. - Konrad Zolna; Serkan Cabi; Yutian Chen 001; Eric Lau; Claudio Fantacci; Jurgis Pasukonis; Jost Tobias Springenberg; Sergio Gomez Colmenarejo (CoRR, 2024)

Mastering diverse control tasks through world models. - Danijar Hafner; Jurgis Pasukonis; Jimmy Ba; Timothy P. Lillicrap (Nat., 2025)