araffin/sbx

RL Environments in JAX 🌍

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

🕹️ A diverse suite of scalable reinforcement learning environments in 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

Monte Carlo tree search in JAX

A library of reinforcement learning components and agents

Multi-Agent Reinforcement Learning with JAX

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)