Top 10 Python atari Projects

tianshou

8 7,378 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.

cleanrl

41 4,414 6.7 Python

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

Project mention: [P] PettingZoo 1.24.0 has been released (including Stable-Baselines3 tutorials) | /r/reinforcementlearning | 2023-08-24

PettingZoo 1.24.0 is now live! This release includes Python 3.11 support, updated Chess and Hanabi environment versions, and many bugfixes, documentation updates and testing expansions. We are also very excited to announce 3 tutorials using Stable-Baselines3, and a full training script using CleanRL with TensorBoard and WandB.

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.
pytorch-a2c-ppo-acktr-gail

3 3,423 0.0 Python

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).
DI-engine

3 2,517 8.8 Python

OpenDILab Decision AI Engine

Project mention: Anyone have experience with DI-Engine? | /r/reinforcementlearning | 2023-05-04

I posted a while back asking people what frameworks they were using for RL research. Recently i stumbled upon DI-Engine which looks promising! Actively maintained, with a diverse set of algorithms already implemented.

LightZero

3 869 8.8 Python

[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios

Project mention: LightZero: Sailing with MCTS, turns the vision of decision intelligence into reality | /r/reinforcementlearning | 2023-06-06

dreamerv2

4 853 0.0 Python

Mastering Atari with Discrete World Models
iris

8 752 1.9 Python

Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%. (by eloialonso)
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.
DQN-Atari

2 64 10.0 Python

Deep Q-Learning (DQN) implementation for Atari pong.

Project mention: Working with DQN ! need some help ! | /r/deeplearning | 2023-05-17

Code Implementation incase this is what you are looking for: https://github.com/KaleabTessera/DQN-Atari

Galaxy-Attack

8 39 0.0 Python

An inspiration of the original Atari Space Invaders game built in pygame :space_invader: :video_game:
pppp

1 3 0.0 Python

Playfield Pixel Perfect Pro: The ultimate Atari 2600 playfield designer

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).