torch>=2.0.0
einops>=0.6.0
numpy>=1.21.0
nvidia-cutlass
ninja
pybind11
psutil
einops>=0.8.0

[dev]
ruff>=0.13.2
jupyterlab>=3.0
pre-commit>=4.0.0
pytest>=8.4.2
pytest-cov>=7.0.0

[docs]
mkdocs
mkdocs-material
mkdocstrings-python
mike
mkdocs-jupyter
mkdocs-redirects
mkdocs-autolinks-plugin
griffe-typingdoc
griffe-inherited-docstrings
griffe
black
mkdocs-same-dir
mdx-breakless-lists
mdx-truly-sane-lists
markdown-gfm-admonition
