numpy
pandas
tqdm
scikit-learn
torch>=1.11.0
torchvision
pybind11
pybind11[global]
matplotlib
mpi4py
