cupy-cuda90 (pypi): CuPy: A NumPy-compatible array library accelerated by CUDA
Keywords: cublas|cuda|cudnn|cupy|curand|cusolver|cusparse|cusparselt|cutensor|gpu|nccl|numpy|nvrtc|nvtx|python|rocm|scipy|tensor