A PyTorch implementation of "Continuous Relaxation Training of Discrete Latent Variable Image Models" - View it on GitHub