Topic Modeling using Variational Auto-Encoders with Gumbel-Softmax and Logistic-Normal Mixture Distributions
Categorical Reparameterization with Gumbel-Softmax