Documentation for mbrl-lib¶
mbrl is library to facilitate research on Model-Based Reinforcement Learning.
mbrl requires Python 3.7+ and PyTorch (>= 1.7).
To install the latest stable version, run
pip install mbrl
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 https://github.com/facebookresearch/mbrl-lib.git pip install -e ".[dev]"
And test it by running
python -m pytest tests/core python -m pytest tests/algorithms
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.