torch>=2.0.0

[cuda]
cuda-toolkit==12.8.2

[dev]
pytest>=7.0
black>=23.0
ruff>=0.1.0
mypy>=1.0

[docs]
mkdocs-material
mkdocstrings[python]
mkdocs-jupyter
