tqdm
numpy
scipy
jinja2
requests
pyparsing
scikit-learn
psutil>=5.8.0

[benchmark]
protobuf<4.21
wandb
pandas
networkx
matplotlib

[dev]
pytest
pytest-cov
onnx
onnxruntime
pre-commit

[full]
yacs
hydra-core
protobuf<4.21
pytorch-lightning
huggingface_hub
ase
h5py
numba
sympy
pandas
captum
rdflib
trimesh
networkx
graphviz
tabulate
matplotlib
torchmetrics
scikit-image
pytorch-memlab
pgmpy
opt_einsum
statsmodels

[graphgym]
yacs
hydra-core
protobuf<4.21
pytorch-lightning

[modelhub]
huggingface_hub

[test]
pytest
pytest-cov
onnx
onnxruntime
