Tianshou: A Highly Modularized Deep Reinforcement Learning Library
Abstract
In this paper, we present Tianshou, a highly modularized Python library for deep reinforcement learning (DRL) that uses PyTorch as its backend. Tianshou intends to be research-friendly by providing a flexible and reliable infrastructure of DRL algorithms. It supports online and offline training with more than 20 classic algorithms through a unified interface. To facilitate related research and prove Tianshou's reliability, we have released Tianshou's benchmark of MuJoCo environments, covering eight classic algorithms with state-of-the-art performance. We open-sourced Tianshou at https://github.com/thu-ml/tianshou/.
Cite
Text
Weng et al. "Tianshou: A Highly Modularized Deep Reinforcement Learning Library." Machine Learning Open Source Software, 2022.Markdown
[Weng et al. "Tianshou: A Highly Modularized Deep Reinforcement Learning Library." Machine Learning Open Source Software, 2022.](https://mlanthology.org/mloss/2022/weng2022jmlr-tianshou/)BibTeX
@article{weng2022jmlr-tianshou,
title = {{Tianshou: A Highly Modularized Deep Reinforcement Learning Library}},
author = {Weng, Jiayi and Chen, Huayu and Yan, Dong and You, Kaichao and Duburcq, Alexis and Zhang, Minghao and Su, Yi and Su, Hang and Zhu, Jun},
journal = {Machine Learning Open Source Software},
year = {2022},
pages = {1-6},
volume = {23},
url = {https://mlanthology.org/mloss/2022/weng2022jmlr-tianshou/}
}