google-deepmind/rlax

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

A library of reinforcement learning components and agents

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

RL Environments in JAX 🌍

Tensorforce: a TensorFlow library for applied reinforcement learning

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

Monte Carlo tree search 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

Python library for Reinforcement Learning.