CuPy: A NumPy-Compatible Library for NVIDIA GPU Calculations. Okuta, R., Unno, Y., Nishino, D., Hido, S., & Loomis, C. In Proceedings of Workshop on Machine Learning Systems (LearningSys) in The Thirty-first Annual Conference on Neural Information Processing Systems (NIPS), 2017.
CuPy: A NumPy-Compatible Library for NVIDIA GPU Calculations [pdf]Paper  abstract   bibtex   
CuPy 1 is an open-source library with NumPy syntax that increases speed by doing matrix operations on NVIDIA GPUs. It is accelerated with the CUDA platform from NVIDIA and also uses CUDA-related libraries, including cuBLAS, cuDNN, cuRAND, cuSOLVER, cuSPARSE, and NCCL, to make full use of the GPU architecture. CuPy’s interface is highly compatible with NumPy; in most cases it can be used as a drop-in replacement. CuPy supports various methods, data types, indexing, broadcasting, and more.

Downloads: 0