Tianshou: A Highly Modularized Deep Reinforcement Learning Library

Jiayi Weng,Huayu Chen,Dong Yan,Kaichao You,Alexis Duburcq,Minghao Zhang,Yi Su,Hang Su,Jun Zhu
DOI: https://doi.org/10.48550/arxiv.2107.14171
2022-01-01
Journal of Machine Learning Research
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/.
What problem does this paper attempt to address?