Experimenting with policy gradient methods in Jax | Hacker News Reader