cupy/cupy: NumPy & SciPy for GPU
Category: math
Language: Python
Topics: cublas, cuda, cudnn, cupy, curand, cusolver, cusparse, cusparselt, cutensor, gpu, nccl, numpy, nvrtc, nvtx, rocm, scipy, tensor
Homepage: https://cupy.dev