torch>=1.4.0
pyyaml
numpy>=1.17

[dev]
black~=22.0
isort>=5.5.4
flake8>=3.8.3
pytest
pytest-xdist

[docs]

[quality]
black~=22.0
isort>=5.5.4
flake8>=3.8.3

[sagemaker]
sagemaker

[test]
pytest
pytest-xdist
