World Models Feature Learning

Experiment with World Models by Ha et al. using Variational Recurrent Neural Networks for more task relevant feature learning.
https://worldmodels.github.io/

VAE and LSTM-MDN derived from https://github.com/dylandjian/retro-contest-sonic and https://github.com/hardmaru/WorldModelsExperiments

VRNN derived from https://gist.github.com/lirnli/c16ef186c75588e705d9864fb816a13c