lucidrains/improving-transformers-world-model-for-rl

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