einops>=0.6
numpy
torch>=1.6
tqdm
