numpy
tqdm
torch

[linting]
pylint
mypy
typing-extensions
pre-commit
ruff
types-tqdm
