beartype
ninja
py-cpuinfo
torch>=2.4
tqdm
typing-extensions>=4.6.0

[dev]
pre-commit
pytest
pytest-xdist
nvtx

[docs]
mkdocs
mkdocs-material
mkdocstrings[python]
mkdocs-static-i18n[material]
matplotlib
