beartype
einops>=0.4
local-attention>=1.0.0
torch>=1.6
vector-quantize-pytorch>=1.1.2
