Also worth mentioning is Jax uses some atari code in their py files. Disambiguation: https://www.cs.toronto.edu/~vmnih/docs/dqn.pdf