mpi4py
numpy
matplotlib
torch
