The Concrete Distribution: A Continuous Relaxation of Discrete Random Variables
Stochastic Variational Inference