numpy
torch
regex
tqdm
