numpy>=2.0.0

[cpu]
torch>=2.8.0

[cuda]

[cuda:python_version >= "3.10" and (sys_platform == "linux" or sys_platform == "win32")]
torch>=2.8.0

[rocm]

[rocm:sys_platform == "linux" and python_version >= "3.10"]
torch>=2.14.0
torchvision
rocm<7.15,>=7.14
