sail-sg/envpool

High throughput synchronous and asynchronous reinforcement learning

A standard API for multi-agent reinforcement learning environments, with popular reference environments and related utilities

A construction kit for reinforcement learning environment management.

A standard API for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

RL Environments in JAX 🌍

An interface library for RL post training with environments.

Multi-Agent Reinforcement Learning with JAX

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

A toolkit for developing and comparing reinforcement learning algorithms.