RL Frameworks

RL Frameworks and Reproducibility

stable-baselines Yes TF 1.14 Atari, MuJoCo ★★★★★ ★★★★★
vel Yes PyTorch 1.0 Atari, MuJoCo ★★★ ★★★★
coach Yes TF Atari, MuJoCo, VizDoom, Fetch ★★★★★ ★★★★★
rllib Yes TF, PyTorch Atari, HalfCheetah ★★★★ ★★★★
Garage Yes TF, PyTorch Pendulum, Walker, HalfCheetah, Fetch, Hopper etc ★★★
anyrl-py TF ★★
lagom Yes PyTorch MuJoCo ★★★ ★★★
lagom Yes PyTorch MuJoCo ★★★ ★★★
reaver Yes TF 1.1+ StarCraft II ★★★ ★★★
TF-Agents TF ★★ ★★
p-christ/DRL PyTorch Four Rooms, Ant etc ★★ ★★★
TorchRL Basic PyTorch Cartpole ★★ ★★
OpenSpiel TF 1.14 games ★★★ ★★★★
bsuite Yes TF 1.13 custom ★★★ ★★★★
dopamine Yes TF Atari ★★★ ★★★★
keras-rl TF ★★★ ★★★
trfl TF ★★★ ★★★
tensorforce basic TF 1.13 ★★★ ★★★
SLM Lab Atari, Roboschool TF 1.13 ★★★ ★★★★
Simple RL TF 1.13 ★★★ ★★★★
Horizon/ReAgent ★★
Torchbeast IMPALA/Atari ★★★

https://winderresearch.com/a-comparison-of-reinforcement-learning-frameworks-dopamine-rllib-keras-rl-coach-trfl-tensorforce-coach-and-more/

Avatar
Shital Shah

A program trying to understand what it’s computing.

comments powered by Disqus