Reinforcement Learning Environments Python Packages

hud-python

RL environments + evals for AI agents. Define once, train anything.

envpool

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

grid2op

Grid2Op a testbed platform to model sequential decision making in power systems.

riichienv

High-Performance Research Environment for Riichi Mahjong

nemo-gym

Evaluate and improve models and agents using environments

esquilax

JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library

pyflyt

UAV Flight Simulator for Reinforcement Learning Research

pynds

PyNDS: A Python Wrapper for the Nintendo DS Emulator

bourse

Rust Market Simulation Library with a Python API

sinergym

Gym environment for building simulation and control using reinforcement learning

bark-ml

Gym environments and agents for autonomous driving.

verbs

Rust and Python Ethereum Agent Based Modelling Library

gym-trading-env

A simple, easy, customizable Gymnasium environment for trading.

gfootball

Check out the new game server:

hrl-pybullet-envs

Locomotion HRL envs in pybullet

gym-cartpole-swingup

A simple, continuous-control environment for OpenAI Gym

sb3-extra-buffers

[ :suspect:👾 ] ➡️ 💾 ➡️ { 🎮🕹️ } Extra Stable-Baselines3 buffer classes. Reducing RL memory usage drastically with minimal overhead.

ai-economist

Foundation: An Economics Simulation Framework

dacbench

A benchmark library for Dynamic Algorithm Configuration.

ns-gym

A framework for modeling non-stationary Markov decision processes and the key decision making problems in these environments

arcle

A Gymnasium-based Environment of the Abstraction and Reasoning Corpus (ARC)

helx

Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env

beobench

A toolkit providing easy and unified access to building control environments for reinforcement learning (RL) [No longer actively maintained]

jaxarc

High-performance JAX-based ARC environment for RL research