vector-quantize-pytorch

这是一个基于PyTorch实现的向量量化（Vector Quantization, VQ）库，支持标量量化和残差VQ。项目的核心功能包括使用指数移动平均更新字典，并提供多种配置选项如`decay`、`commitment_weight`等参数以适应不同需求。此外，还实现了ResidualVQ类，允许用户通过递归地量化波形残差来提高模型性能，适用于需要高质量图像生成或音乐合成的应用场景，如VQ-VAE-2和Jukebox项目所示。该库简洁易用，适合深度学习研究者及开发者在进行相关领域探索时采用。