Reinforcement Learning Algorithms

Stable Baselines3Python

A reliable PyTorch implementation of reinforcement learning algorithms for research and industry.

rex-gymPython

OpenAI Gym environments and tools for training a 3D-printed quadruped robot (SpotMicro) using reinforcement learning.

purejaxrlPython

High-performance, end-to-end reinforcement learning implementations fully written in JAX for massive parallelization on GPUs.