Best PyTorch RL library for doing research

• cleanrl

  41 4,459 6.3 Python

  High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

cleanrl is the repo I base my code on, lots of single file implementations.

• sample-factory

  6 740 8.1 Python

  High throughput synchronous and asynchronous reinforcement learning

I borrow a lot of performance tricks from sample factory, which is awesome but hard to modify from its original APPO algorithm. rlpyt was more modular, and I borrowed more ideas from it (namedarraytuple), but still too limited.

• WorkOS

  workos.com sponsored

  The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  

• rlpyt

  4 2,197 0.0 Python

  Reinforcement Learning in PyTorch

I borrow a lot of performance tricks from sample factory, which is awesome but hard to modify from its original APPO algorithm. rlpyt was more modular, and I borrowed more ideas from it (namedarraytuple), but still too limited.

• tianshou

  8 7,406 9.5 Python

  An elegant PyTorch deep reinforcement learning library.

  

I tried tianshou and thought it was well-designed for modularity, but it was early in development when I tried and missing some basic features

• stable-baselines3

  46 7,894 8.2 Python

  PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

  

stable baselines 3 is really starting to shape up, but I personally think it's better as a reference guide, there's a bit too much abstraction for the sake of ease of first plot, I wouldn't want to have to modify it.

• Deep-Reinforcement-Learning-Algorithms-with-PyTorch

  2 5,416 3.6 Python

  PyTorch implementations of deep reinforcement learning algorithms and environments

• mbrl-lib

  1 904 3.4 Python

  Library for Model Based RL

  

MBRL-Lib for model-based RL

• InfluxDB

  www.influxdata.com sponsored

  Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

  

• mtrl

  1 211 0.0 Python

  Discontinued Multi Task RL Baselines

  

MTRL for multi-task RL

• machin

  2 381 1.8 Python

  Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...

Machin is really nice, it is very easy to use and to try different things, although it’s developed by one person and maybe not appropriately tested yet.

Suggest a related project