Dopamine logo

Dopamine is a framework for flexible, value-based, reinforcement learning research. It was originally written in TensorFlow, but now all agents have been implemented in JAX.

You can read more about it in our github page and in our white paper.

Original Google AI blogpost.

We have a website where you can easily compare the performance of all the Dopamine agents, which I find really useful:

Baselines screenshot .

We also provide a set of Colaboratory notebooks that really help understand the framework: