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