Unbiased scalable softmax optimization
The Concrete Distribution: A Continuous Relaxation of Discrete Random Variables