A Gentle Introduction to Deep Reinforcement Learning in JAX

A Gentle Introduction to Deep Reinforcement Learning in JAX

Solving the CartPole environment with DQN in under a second