ETH Zürich; ETH Zürich, MPI for Intelligent Systems, Tübingen; University of Toronto, Vector Institute
Max B Paulus, Chris Maddison, Andreas Krause
gumbel;softmax;gumbel-softmax;straight-through;straightthrough;rao;rao-blackwell
Rao-Blackwellizing the Straight-Through Gumbel-Softmax Gradient Estimator