Pytorch implementation of the straight-through gumbel-rao estimator from "Rao-Blackwellizing the Straight-Through Gumbel-Softmax Gradient Estimator" https://arxiv.org/abs/2010.04838.
Pytorch implementation of the straight-through gumbel-rao estimator from "Rao-Blackwellizing the Straight-Through Gumbel-Softmax Gradient Estimator" https://arxiv.org/abs/2010.04838.