Probabilistic Adaptive Computation Time
The Concrete Distribution: A Continuous Relaxation of Discrete Random Variables