numpy
torch>=2.2.0
typing
triton>=2.2.0
networkx
numba
packaging
matplotlib
tqdm

[dev]
pytest
pytest-xdist
pytest-skip-slow
torchvision
datasets
matplotlib
