torch>=2.0.0
cuda-python>=12.0.0
numpy

[dev]
cupy>=12.0.0
pytest>=7.0.0
build>=0.10.0
