beartype
einops>=0.7.0
vector-quantize-pytorch>=1.9.7
torch
