# world-models-pytorch
My attempt to implement "World Models" by David Ha and Jurgen Schmidhuber

## The work is in progress


TODOs

-[x] Implement MDN-RNN
-[ ] Implement VAE
-[ ] Training for VAE
    -[ ] Random rollouts to create a dataset
    -[ ] Training function
-[ ] Training for MDN-RNN
-[ ] Evolution strategy for training Controller