A MNIST example of variational auto encoder in TensorFlow/Keras by TPUs.
Random numbers for sampling and reparameterization trick, and KL divergence of the middle layer are connected by skip-connection, respectively.
This makes it possible to define VAEs in one model, making it easier to train with TPU.