Building Agents with Imagination: pytorch step-by-step implementation - View it on GitHub