https://lilianweng.github.io/posts/2018-04-08-policy-gradient/