https://byrzhm.github.io/blog/posts/pytorch-internals/