Inverse reinforcement learning in contextual MDPs. - Belogolovsky, Stav; Korsunsky, Philip; Mannor, Shie; Tessler, Chen; Zahavy, Tom (Machine Learning, 2021)