Reinforcement learning libraries

31 libraries, 17 updated this year

Libraries for agents that learn by acting in an environment. Gymnasium, the Farama Foundation's maintained continuation of OpenAI Gym, defines the standard environment interface. Stable-Baselines3 implements reliable versions of common algorithms such as PPO, SAC and DQN in PyTorch, and RLlib scales training across a cluster on Ray.

For learning the field, OpenAI's Spinning Up pairs short implementations with explanations. Research platforms such as DeepMind Lab, ViZDoom and Gym Retro provide 3D and game environments, and DeepMind's acme collects agent building blocks. Vowpal Wabbit covers contextual bandits, which are often the practical form of reinforcement learning in production.

Several environments here were archived when their maintainers moved on, OpenAI's Roboschool and Gym Retro among them; the activity column marks them.

31 libraries

RLlibray-project/ray

RLlib is an industry level, highly scalable RL library for tf and torch, based on Ray. It's used by companies like Amazon and Microsoft to solve real-world decision making problems at scale.

PythonDeep learning
44.1k updated today

ML-From-Scratcheriklindernoren/ML-From-Scratch

Implementations of Machine Learning models from scratch in Python with a focus on transparency. Aims to showcase the nuts and bolts of ML in an accessible way.

PythonGeneralDeep learning
33.1k updated 3y ago

stable-baselines3DLR-RM/stable-baselines3

PyTorch implementations of Stable Baselines (deep) reinforcement learning algorithms.

Go
13.9k updated 11d ago

GymnasiumFarama-Foundation/Gymnasium

A library for developing and comparing reinforcement learning algorithms (successor of [gym])(https://github.com/openai/gym).

Python
12.7k updated today

Spinning Upopenai/spinningup

An educational resource designed to let anyone learn to become a skilled practitioner in deep reinforcement learning

Python
12k updated 2y ago

Weights & Biaseswandb/wandb

Machine learning experiment tracking, dataset versioning, hyperparameter search, visualization, and collaboration

Deep learningData & vizMLOps
11.3k updated today

Vowpal Wabbit (VW)VowpalWabbit/vowpal_wabbit

A fast out-of-core learning system.

C++General
8.7k updated 9d ago

DeepMind Labdeepmind/lab

DeepMind Lab is a 3D learning environment based on id Software's Quake III Arena via ioquake3 and other open source software. Its primary purpose is to act as a testbed for research in artificial intelligence, especially deep reinforcement learning.

PythonDeep learning
7.4k updated 3y ago

Serpent.AISerpentAI/SerpentAI Archived

Serpent.AI is a game agent framework that allows you to turn any video game you own into a sandbox to develop AI and machine learning experiments. For both researchers and hobbyists.

PythonDeep learningVision
7k archived

acmegoogle-deepmind/acme

An Open Source Distributed Framework for Reinforcement Learning that makes build and train your agents easily.

Python
4.1k updated 1d ago

Polyaxonpolyaxon/polyaxon

A platform for reproducible and scalable machine learning and deep learning.

C++GeneralDeep learning
3.7k updated 1d ago

DI-engineopendilab/DI-engine

DI-engine is a generalized Decision Intelligence engine. It supports most basic deep reinforcement learning (DRL) algorithms, such as DQN, PPO, SAC, and domain-specific algorithms like QMIX in multi-agent RL, GAIL in inverse RL, and RND in exploration problems.

Python
3.6k updated 10mo ago

Retroopenai/retro Archived

Retro Games in Gym

Python
3.6k archived

TensorWatchmicrosoft/tensorwatch

Debugging and visualization tool for machine learning and data science. It extensively leverages Jupyter Notebook to show real-time visualizations of data in running processes such as machine learning training.

PythonDeep learningData & viz
3.5k updated 6mo ago

Catalystcatalyst-team/catalyst

High-level utils for PyTorch DL & RL research. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusing. Being able to research/develop something new, rather than write another regular train loop.

PythonGeneralDeep learningVision
3.4k updated 3mo ago

Roboschoolopenai/roboschool Archived

Open-source software for robot simulation, integrated with OpenAI Gym.

Python
2.2k archived

garagerlworkgroup/garage

A toolkit for reproducible reinforcement learning research

PythonDeep learning
2.1k updated 3y ago

ViZDoommwydmuch/ViZDoom

ViZDoom allows developing AI bots that play Doom using only the visual information (the screen buffer). It is primarily intended for research in machine visual learning, and deep reinforcement learning, in particular.

PythonDeep learning
2.1k updated today

metaworldrlworkgroup/metaworld

An open source robotics benchmark for meta- and multi-task reinforcement learning

Python
1.9k updated 15d ago

SLM Labkengz/SLM-Lab

Modular Deep Reinforcement Learning framework in PyTorch.

PythonDeep learning
1.4k updated 1mo ago

RLtoolsrl-tools/rl-tools

The fastest deep reinforcement learning library for continuous control, implemented header-only in pure, dependency-free C++ (Python bindings available as well).

C++Deep learning
1k updated 1d ago

Pavlov.jsNathanEpstein/Pavlov.js

Reinforcement learning using Markov Decision Processes.

JavaScriptGeneral
497 updated 8y ago

goldaunum/gold

A reinforcement learning library.

Go
352 updated 6y ago

Mazeenlite-ai/maze

Application-oriented deep reinforcement learning framework addressing real-world decision problems.

PythonDeep learning
294 updated 4mo ago

evostraalirezamika/evostra

A fast Evolution Strategy implementation in Python.

PythonGeneralDeep learning
271 updated 6y ago

BanditLibjkomiyama/banditlib Deprecated

A simple Multi-armed Bandit library.

C++General
139 updated 2y ago

neatjinyeom/neat Archived

Plug-and-play, parallel Go framework for NeuroEvolution of Augmenting Topologies (NEAT).

GoGeneralDeep learning
73 archived

Gym4ReaLDaveonwave/gym4ReaL

Gym4ReaL is a comprehensive suite of realistic environments designed to support the development and evaluation of RL algorithms that can operate in real-world scenarios. The suite includes a diverse set of tasks exposing RL algorithms to a variety of practical challenges.

Python
48 updated 1y ago

AI on Browserai-on-browser/ai-on-browser.github.io

An educational, pure JavaScript library designed to help developers and students understand the inner workings of ML algorithms without the magic of external libraries.

JavaScriptGeneral
19 updated today

Bread WandB ViewerBread-Technologies/bread_wandb_viewer_extension

A VS Code extension to view Weights & Biases experiments, logs, and artifacts within the IDE, eliminating the need to switch to the web UI and keeping data private.

MLOps
8 updated 8mo ago

CoachNervanaSystems/coach

Reinforcement Learning Coach by Intel® AI Lab enables easy experimentation with state of the art Reinforcement Learning algorithms

Python
No GitHub repo
Esc

Type to search every entry. ↑↓ to move, Enter to open.