ericjang/gumbel-softmax

categorical variational autoencoder using the Gumbel-Softmax estimator

gumbel-softmax

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