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.

di-engine

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

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

mbrl

Library for Model Based RL

bellman

Model-based reinforcement learning in TensorFlow

deepcubeai

Learning Discrete World Models for Heuristic Search (RLC 2024)

baconian

Model-based Reinforcement Learning Framework