nvidia-ml-py
typer
rich>=13.8.0
torch

[dev]
coverage
mypy
pytest
ruff
black
pre-commit
bump-my-version
setuptools<82,>=77

[macm]
psutil

[rocm]
