Vector Quantized VAEs - PyTorch Implementation - View it on GitHub