Model Based Reinforcement Learning Python Packages

torchrl

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

torchrl-nightly

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

dreamer4

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

improving-transformers-world-model

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

di-engine

OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.

mbrl

Library for Model Based RL

bellman

Model-based reinforcement learning in TensorFlow

deepcubeai

Learning Discrete World Models for Heuristic Search

baconian

model-based reinforcement learning toolbox