torch>=2.0
numpy>=1.24

[cuda]
pycuda
cuda-python

[dev]
pytest>=7.0
pytest-benchmark
