Documentation for mbrl-lib

mbrl is library to facilitate research on Model-Based Reinforcement Learning.

Getting started


Standard Installation

mbrl requires Python 3.7+ and PyTorch (>= 1.7).

To install the latest stable version, run

pip install mbrl

Development Installation

If you are interested in modifying parts of the library, you can clone the repository and set up a development environment, as follows

git clone
pip install -e ".[dev]"

And test it by running

python -m pytest tests/core
python -m pytest tests/algorithms

Basic Example

As a starting point, check out our tutorial notebook on how to write the PETS algorithm (Chua et al., NeurIPS 2018) using our toolbox, and running it on a continuous version of the cartpole environment. Then, please take a look at our API documentation below.