Context is Environment - Gupta, Sharut; Jegelka, Stefanie; Lopez-Paz, David; Ahuja, Kartik (International Conference on Learning Representations, 2024)