beartype
einops>=0.7.0
vector-quantize-pytorch>=1.9.10
torch
torchvision
