cupy-cuda12x (pypi): CuPy: NumPy & SciPy for GPU
Keywords: cublas|cuda|cudnn|cupy|curand|cusolver|cusparse|cusparselt|cutensor|gpu|nccl|numpy|nvrtc|nvtx|python|rocm|scipy|tensor