RobertTLange/gymnax

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

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

SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms

Multi-Agent Reinforcement Learning with JAX

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

A toolkit for developing and comparing reinforcement learning algorithms.

Isaac Gym Reinforcement Learning Environments

An interface library for RL post training with environments.