Categorical Reparameterization with Gumbel-Softmax
Gradient Estimation Using Stochastic Computation Graphs