PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution - View it on GitHub