Shypd.aiThe largest open index of AI toolsTianshou is an elegant PyTorch deep reinforcement learning library that provides modular low-level interfaces for algorithm developers and convenient high-level interfaces for applications of RL.
Trending
About
Tianshou (夊ć) is a reinforcement learning (RL) library built on pure PyTorch and Gymnasium, designed for both RL researchers and practitioners. It offers modular low-level interfaces for algorithm developers, emphasizing flexibility, hackability, and type-safety, alongside convenient high-level interfaces for applying RL to custom environments. The library supports a wide range of RL paradigms, including online (on- and off-policy) and offline RL, with experimental support for multi-agent RL (MARL) and model-based RL. Tianshou stands out for its high-performance, modularized framework, user-friendly interfaces, and generality, enabling concise implementations without sacrificing flexibility. It includes implementations of numerous state-of-the-art algorithms like DQN, PPO, SAC, and various offline RL methods, and supports features such as vectorized environments, recurrent state representations, and multi-GPU training.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
