AMAGO-2: Breaking the Multi-Task Barrier in Meta-Reinforcement Learning with Transformers. - Jake Grigsby; Justin Sasek; Samyak Parajuli; Daniel Adebi; Amy Zhang 001; Yuke Zhu (NeurIPS, 2024)