Gumbel-Softmax-Implementation-with-VAE The implementation of Gumbel softmax reparametrization trick for discrete VAE