Vector Quantization - Pytorch

A vector quantization library for PyTorch, originally transcribed from Deepmind's TensorFlow implementation. It focuses on using exponential moving averages to update the dictionary and has been applied successfully in generative models for images (VQ-VAE-2) and music (Jukebox). The library is actively maintained with frequent micro-releases, often incorporating new research techniques.