Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE) - View it on GitHub