categorical variational autoencoder using the Gumbel-Softmax estimator - View it on GitHub