Monte Carlo tree search in JAX
Advanced evolutionary computation library built on top of PyTorch
A Modular Simulation Framework and Benchmark for Robot Learning
A collection of reference Jupyter notebooks and demo AI/ML application
Training framework for Stable Baselines3 reinforcement learning agents
SAPIEN Manipulation Skill Framework
RL implementations
A high-performance distributed training framework
A library of reinforcement learning components and agents
Modular Deep Reinforcement Learning framework in PyTorch