einops>=0.4
vector-quantize-pytorch>=0.10.14
torch>=1.6
