Transformer Variational Autoencoder experiment - View it on GitHub