CuPy
General-purpose Matrix Calculation Library for GPU
Overview
CuPy™ is an open-source matrix library accelerated with NVIDIA CUDA. It has a NumPy-compatible interface and uses CUDA-related libraries including cuBLAS, cuDNN, cuRand, cuSolver, cuSPARSE, cuFFT, and NCCL to make full use of the GPU architecture.
CuPy was born as a GPU backend of Chainer, and it has been separated from Chainer into an independent package from 2017.
Features
NumPy compatible GPU library for fast computation in Python
Pre-built binaries for fast installation
Projects leveraging CuPy
CuPy: A NumPy Compatible Library for High Performance Computing with GPU | SciPy 2019
