Categorical Reparameterization with Gumbel-Softmax – PaperGrep

https://papergrep.dev/paper/categorical-reparameterization-with-gumbel-softmax-d6d824