timm
torch>1.10.0
torchvision>0.11.1
mpi4py