torch
numpy

[dev]
ruff
mypy
pytest
timm
torchvision
matplotlib
tqdm
ipywidgets

[test]
pytest
timm
