numpy
numba
cuda-bindings
cuda-core
numba-cuda
cuda-cccl
cupy
nvidia-cute-dsl[cu13]
torch[cu13]