Fetching the paper…

Rao-Blackwellizing the Straight-Through Gumbel-Softmax Gradient Estimator · Around