numpy
tqdm
scipy
scikit-learn
ipython
torch==1.5.0
torchvision==0.6.0
transformers==2.11.0