tianshou

Tianshou 是一个基于 PyTorch 的深度强化学习库。它提供了灵活且类型安全的低级接口供算法开发者使用，同时也为应用开发者准备了便捷的高级接口来训练自定义环境中的已有算法。该库支持在线（包括on-policy和off-policy）和离线RL，并实验性地支持多智能体RL、基于模型的RL等。Tianshou 以其高性能、模块化设计以及友好的用户界面而著称，非常适合需要快速原型开发或研究新算法的研究人员及工程师在各种RL应用场景中使用。