numpy
pandas
scipy
scikit-learn
torch
torch-geometric
networkx

[dev]
pytest
pytest-cov
