Our research on improving zero-shot generalization of world models for Contextual RL has been accepted at the inaugural Reinforcement Learning Conference (RLC). You can find the preprint of the paper here. You should also check out Sai’s twitter thread on the paper here.