Gumbel-Softmax version of Bernoulli and Categorical distributions

Does pymc have gumbel-softmax reparametrization of Bernoulli and Categorical?
Does that reparametrization actually improve fitting of Bernoulli and Categorical?
Is it possible to use a random variable for the temperature?

Thanks again.

1 Like