numpy
scipy
torch==1.7.1
tqdm
networkx
transformers
click
ipython
torchtext==0.8.1
seaborn
matplotlib
tensorboard

[testing]
pytest
pytest-cov
