torch
torchvision

[dev]
pre-commit==2.15.0
toml==0.10.2
pytest==6.2.5
ipywidgets
jupyterlab
