AMAGO: Scalable In-Context Reinforcement Learning for Adaptive Agents - Grigsby, Jake; Fan, Jim; Zhu, Yuke (International Conference on Learning Representations, 2024)