numpy
scipy
matplotlib
pandas
seaborn
tqdm
mpi4py
scikit-learn
h5py>=3.15.1

[gpu]
torch
