torch>=2.0.1
numpy
tqdm

[all]
black
flake8
isort
coverage
pytest

[dev]
black
flake8
isort
coverage
pytest

[lint]
black
flake8
isort

[test]
coverage
pytest
