torch>=1.8.0
torchvision>=0.9.0
tqdm
opt_einsum
einops
ftfy
regex
