https://venkat-rajgopal.github.io/posts/rectified-adam-optimizer/