model-based-reinforcement-learning

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

Implementation of Danijar's latest iteration for his Dreamer line of work

Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch