einops
numpy
matplotlib
opencv-python
h5py
torch
torchvision
transformers==3.5.1
gensim
tensorflow
captum
ipython
wget