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
scikit-image
opencv-python
timm
torchvision
torchmetrics
torchvision
torch
torch-geometric

[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
opencv-python
timm
torchvision
torch
torch-geometric

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

[modelhub]
huggingface_hub

[test]
pytest
pytest-cov
onnx
onnxruntime
