torch>=2.0.0

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

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