Top 7 Python policy-gradient Projects

tianshou

8 7,406 9.5 Python

An elegant PyTorch deep reinforcement learning library.

Project mention: Is it better to not use the Target Update Frequency in Double DQN or depends on the application? | /r/reinforcementlearning | 2023-07-05

The tianshou implementation I found at https://github.com/thu-ml/tianshou/blob/master/tianshou/policy/modelfree/dqn.py is DQN by default.

PPO-PyTorch

2 1,453 2.8 Python

Minimal implementation of clipped objective Proximal Policy Optimization (PPO) in PyTorch
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.
HandyRL

1 282 4.3 Python

HandyRL is a handy and simple framework based on Python and PyTorch for distributed reinforcement learning that is applicable to your own environments.
pytorch-learn-reinforcement-learning

3 139 0.0 Python

A collection of various RL algorithms like policy gradients, DQN and PPO. The goal of this repo will be to make it a go-to resource for learning about RL. How to visualize, debug and solve RL problems. I've additionally included playground.py for learning more about OpenAI gym, etc.
episodic-transformer-memory-ppo

5 108 0.0 Python

Clean baseline implementation of PPO using an episodic TransformerXL memory

Project mention: Question about Transformer model input in RL | /r/reinforcementlearning | 2023-06-17

Check out this implementation https://github.com/MarcoMeter/episodic-transformer-memory-ppo

nes-torch

3 17 3.6 Python

Minimal PyTorch Library for Natural Evolution Strategies
pbo

1 13 3.0 Python

Policy-based optimization : single-step policy gradient seen as an evolution strategy
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.

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).