torch
numpy
mpi4py
scikit-learn
prettyPlot>=0.0.27
