numpy
pandas
torch
tqdm
ipython
graphviz

[dev]
ruff
pytest
pre-commit

[test]
lightning
pillow
pytest
requests
timm
torchaudio
torch_geometric
torchvision
transformers
visualpriors
