vector-quantize-pytorch