training language models to reason with a world model: https://laion.ai/notes/rook/