Probabilistic Adaptive Computation Time
Categorical Reparameterization with Gumbel-Softmax