Categorical Reparameterization with Gumbel-Softmax
Variational inference for Monte Carlo objectives