Implementation of the Gumbel-Sigmoid distribution in PyTorch.
pytorch
gumbel-softmax
gumbel-distribution
reparameterization-trick
straight-through-gradient
gumbel-sigmoid
-
Updated
Jul 22, 2022 - Python