Differentiable Sampling of Categorical Distributions Using the CatLog-Derivative Trick

Neural Information Processing Systems 

Our first contribution addresses this shortcoming by introducing the CatLog-Derivative trick-a variation of the Log-Derivative trick tailored towards categorical distributions.