numpy>=1.26.4
# cupy-cuda12x>=12.0.0 # Optional, for CUDA support