nvidia-ml-py
typer
torch
colorlog

[dev]
coverage
mypy
pytest
ruff
black
pre-commit
bump-my-version
