Atari Python Packages

tianshou

An elegant PyTorch deep reinforcement learning library.

aftab

Combining advancements in discrete deep reinforcement learning

di-engine

OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.

lightzero

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

crosslearn

Reusable representation extractors for reinforcement learning across vectors, images, and Chronos-backed time-series windows.

keras-rl2

Reinforcement learning with tensorflow 2 keras

mushroom-rl

A Python library for Reinforcement Learning experiments.

openrl

unified reinforcement learning framework

xuance

XuanCe: A Comprehensive and Unified Deep Reinforcement Learning Library

dreamerv2

Mastering Atari with Discrete World Models

sb3-extra-buffers

Extra buffer classes for Stable-Baselines3, reduce memory usage with minimal overhead.

adeptrl

Reinforcement learning framework to accelerate research

pyreinforce

Deep Reinforcement Learning library for Python

gxm

Reinforcement Learning Environments in JAX.

dqn-ale-spaceinvaders

A Deep Q-Network (DQN) implementation for Atari Space Invaders using Gymnasium and PyTorch.

gym-games

A collection of Gymnasium compatible games for reinforcement learning.

toybox-api

The Machine Learning Toybox for testing the behavior of autonomous agents.

toybox-envs

The Machine Learning Toybox for testing the behavior of autonomous agents.

worldflux

Unified Python interface for world models in reinforcement learning

toybox-cffi

The Machine Learning Toybox for testing the behavior of autonomous agents.