numpy
torch
tqdm