Categorical Reparameterization with Gumbel-Softmax

   Abstract