torch>=2.0
nvidia-cutlass<4,>=3.5

[bench]
torchvision

[docs]
sphinx>=7
furo
myst-parser

[test]
pytest
