/gumbel-softmax-pytorch

categorical variational autoencoder using the Gumbel-Softmax estimator

Primary LanguageJupyter Notebook