Implementation of the world model architecture for self driving out of Wayve - View it on GitHub