Categorical Reparameterization with Gumbel-Softmax
InfoGAN: Interpretable Representation Learning by Information Maximizing Generative Adversarial Nets