torch>=2.0.0

[cupy]

[cupy:platform_machine != "arm64"]
cupy-cuda12x>=12.0.0

[cupy:platform_machine == "arm64"]
cupy>=12.0.0
