Gradient Estimation with Stochastic Softmax Tricks - Paulus, Max; Choi, Dami; Tarlow, Daniel; Krause, Andreas; Maddison, Chris (Advances in Neural Information Processing Systems, 2020)