Adds working transformer world model

Merge pull request #8 from mauicv/feature/add-discrete-states

Feature/add discrete states

Merge pull request #7 from mauicv/feature/value-trainer

Feature/value trainer

Merge pull request #6 from mauicv/feature/add-rssm-world-model

Add PlaNet RSSM world model