mpi4py>=3.0.0
scikit-learn>=1.0.1
numpy>=1.21.4
