tinker495/jax-baseline

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

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

A framework for Reinforcement Learning research.

RL Environments in JAX 🌍

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

Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code

A library of reinforcement learning components and agents

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