torch
lightning
torch_geometric
matplotlib

[dev]
black

[doc]
sphinx<8.2,>5.0
sphinx_rtd_theme
sphinx_copybutton
sphinx_design
pydata_sphinx_theme

[test]
pytest
pytest-cov
scipy

[tutorial]
jupyter
smithers
torchvision
tensorboard
scipy
numpy
