pytorch>=1.4.0
openmpi==4.0.3
mpi4py==3.0.3
numpy==1.18.4
h5py==2.10.0
matplotlib==3.2.1
scipy==1.4.1
scikit-learn==0.22.2.post1
seaborn==0.10.1
