vector_quantize_pytorch
