Before revision:  IMAGINE learns to represent goals by jointly learning a language model and a goal-conditioned reward function.
After revision:  IMAGINE learns to represent goals by jointly learning a language encoder and a goal-conditioned reward function.