Categorical Reparameterization with Gumbel-Softmax