JAX Implementations of Actor-Critic Algorithms

• stable-baselines3

  46 8,115 8.2 Python

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

  

I even have the PyTorch implementation faster in some cases (I created a branch with pytorch optimization that gives a 5% speed improvement https://github.com/DLR-RM/stable-baselines3/tree/exp/torch-optim ).

• Ray

  43 31,566 10.0 Python

  Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

  

Folks like me using RLLib have observed this behavior: https://github.com/ray-project/ray/issues/12494

• Scout Monitoring

  www.scoutapm.com featured

  Free Django app performance insights with Scout Monitoring. Get Scout setup in minutes, and let us sweat the small stuff. A couple lines in settings.py is all you need to start monitoring your apps. Sign up for our free tier today.

  

• stable-baselines

  10 4,068 0.0 Python

  A fork of OpenAI Baselines, implementations of reinforcement learning algorithms

- tf2 speed: https://github.com/hill-a/stable-baselines/issues/576#issuecomment-573331715

• rl-baselines3-zoo

  11 1,824 6.2 Python

  A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

  

for pytorch, use the rl zoo (https://github.com/DLR-RM/rl-baselines3-zoo) and sb3 ;) https://github.com/DLR-RM/stable-baselines3

Suggest a related project