colorama>=0.4.6
numpy
pytest>=9.1.1

[cuda]
cupy
