vqvae. A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)