numpy>=1.22

[cuda]
nvidia-cuda-nvrtc-cu12
nvidia-cuda-runtime-cu12

[cuda11]
nvidia-cuda-nvrtc-cu11
nvidia-cuda-runtime-cu11

[cuda12]
nvidia-cuda-nvrtc-cu12
nvidia-cuda-runtime-cu12

[cuda13]
nvidia-cuda-nvrtc<14,>=13
nvidia-cuda-runtime<14,>=13

[jax]
jax>=0.5

[rocm]
rocm-sdk-core

[test]
pytest
torch

[torch]
torch>=2.4

[wheeltest]
pytest
jax>=0.5
