This is the code for "World Models Explained" By Siraj Raval on Youtube - View it on GitHub