This is a python JAX implementation of the paper: Rainbow: Combining improvements in deep reinforcement learning, by M. Hessel et al. In Thirty-Second AAAI Conference on Artificial Intelligence.
reinforcement-learning
deep-learning
deep-reinforcement-learning
dqn
haiku
jax
papers-with-code
dm-haiku
dqn-jax
-
Updated
Jul 20, 2020 - Jupyter Notebook