torch
torchvision

[dev]
black
isort
pytest

[publish]
build
twine
