numpy
torch
transformers
timm
matplotlib
albumentations
tqdm