numpy>=1.21.0
psutil>=5.8.0
pyyaml>=6.0

[all]
torch>=2.0.0
nvidia-ml-py>=11.450.51
intel-extension-for-pytorch>=2.0.0
docker>=6.0.0

[amd]
torch>=2.0.0

[dev]
pytest>=7.0
black>=22.0
flake8>=4.0
mypy>=0.900
build>=0.10.0
twine>=4.0.2

[intel]
torch>=2.0.0
intel-extension-for-pytorch>=2.0.0

[nvidia]
torch>=2.0.0
nvidia-ml-py>=11.450.51
