discrete-variables
Relaxit: A Python Library for Optimizing Discrete Probability Distributions in Neural Networks