torch>=2.0.0
pydantic>=2
torch_geometric>=2.4.0
numpy<2
scikit-learn
scipy
filelock
pyyaml>=6
pytorch-minimize
