This is a VAE. While tf probability is a library that can help implement a VAE, I chose to implement it with the simplest building blocks (i.e. bare tf), so it’s easier to understand.
If you want to learn more about VAE so it’s easier to understand why my implementation is indeed a VAE, you’re welcome to read my previous post.