How I Implemented PPO in JAX to Score 300 at Atari Breakout

Proximal Policy Optimization (PPO) is a reinforcement learning algorithm, introduced by John Schulman and colleagues at OpenAI in 2017. It…