gumbel-softmax categorical variational autoencoder using the Gumbel-Softmax estimator. Paper is here: https://arxiv.org/abs/1611.01144