einops
h5py
mpi4py
numpy
tensorboard
torch
tqdm