https://aestheticvoyager.github.io/aesvoy/posts/variational-auto-encoder/