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