gumbel-softmax
Relaxit: A Python Library for Optimizing Discrete Probability Distributions in Neural Networks