Meta reinforcement learning as task inference