gumbel_sigmoid A pytorch implementation of the gumbel sigmoid gate (paper) example logits = torch.randn(1000) out = gumbel_sigmoid(logits, temp=0.6)