MBRL-Lib: A Modular Library for Model-based Reinforcement Learning

  • 2021-04-20 17:58:22
  • Luis Pineda, Brandon Amos, Amy Zhang, Nathan O. Lambert, Roberto Calandra
  • 27

Abstract

Model-based reinforcement learning is a compelling framework fordata-efficient learning of agents that interact with the world. This family ofalgorithms has many subcomponents that need to be carefully selected and tuned.As a result the entry-bar for researchers to approach the field and to deployit in real-world tasks can be daunting. In this paper, we present MBRL-Lib -- amachine learning library for model-based reinforcement learning in continuousstate-action spaces based on PyTorch. MBRL-Lib is designed as a platform forboth researchers, to easily develop, debug and compare new algorithms, andnon-expert user, to lower the entry-bar of deploying state-of-the-artalgorithms. MBRL-Lib is open-source athttps://github.com/facebookresearch/mbrl-lib.

 

Quick Read (beta)

loading the full paper ...