torch>=2.0.0
torch_geometric>=2.3.0

[benchmark]
ogb
pyyaml

[dev]
pytest
pytest-cov
build
twine
pyyaml
ruff
mypy
pre-commit

[full]
faiss-cpu
umap-learn
matplotlib
seaborn
ogb
pyyaml

[viz]
umap-learn
matplotlib
seaborn
