google / dopamine

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

Home Page:https://github.com/google/dopamine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Diffenence against Rlax

JiahaoYao opened this issue · comments

Hi, there are currently two lib for RL using Jax, Dopamine and Rlax. Might I ask what is the difference? Thanks!

Thanks @psc-g and great episode on talkRL https://www.talkrl.com/episodes/pablo-samuel-castro 👍

... knowledge on distributional RL, bisimulation, the Dopamine RL Framework, TF-Agents, and much more!

I understand that, unlike Dopamine (Google Research), RLax has DeepMind-built dependencies like Haiku (similar to Sonnet in TensorFlow) and Optax.

Do you know if both libraries are meant for fast prototyping? The Dopamine example on CartPole with DQN and C51 appears to have less imports from the get-go.

Thanks!