torch>=1.10.0
typing_extensions

[dev]
black==23.12.1
coverage
flake8
ipython
pre-commit
pytest
torchvision>=0.11.0
twine

[extras]
h5py
numpy
tensorboard
tqdm
