# world-models-pytorch
My attempt to implement "World Models" by David Ha and Jurgen Schmidhuber
Link: https://arxiv.org/pdf/1803.10122.pdf
## 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