dopamax (pypi): Reinforcement learning in pure JAX.
Keywords: anakin|dopamax|dqn|jax|podracer|ppo|reinforcement-learning