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