torch-geometric>=2.7.0
torch
scikit-learn
numpy
scipy
matplotlib
pyyaml
ogb

[benchmark]
ogb

[dev]
pytest
sphinx>=6.0
sphinx_rtd_theme
nbsphinx
myst_parser
sphinx-copybutton
sphinx-autodoc-typehints
black
flake8
isort
mypy
pre-commit

[docs]
sphinx>=6.0
sphinx_rtd_theme
nbsphinx
myst_parser
sphinx-copybutton
sphinx-autodoc-typehints

[test]
pytest

[triton]
triton
