torch>=2.0.0

[cuda]
cuda-toolkit==12.8.2

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