https://dave90.github.io/posts/ml-vae/