click>=8.0.0
psutil>=5.9.0
pydantic>=2.0.0
nvidia-ml-py>=12.0.0

[:python_version < "3.11"]
backports.strenum>=1.3.1

[dev]
pytest>=7.0.0
black>=22.0.0
ruff>=0.1.0
pre-commit>=3.0.0
mypy>=1.0.0
types-psutil>=5.9.0
types-PyYAML

[stress]
torch>=2.0.0
numpy>=1.24.0
