CuPy

[wtm_mlop_cats]

Features

CuPy is an implementation of NumPy-compatible multi-dimensional array on CUDA. CuPy consists of cupy.ndarray, the core multi-dimensional array class, and many functions on it. It supports a subset of numpy.ndarray interface.

Basic indexing (indexing by ints, slices, newaxes, and Ellipsis)

Most of Advanced indexing (except for some indexing patterns with boolean masks)

Data types (dtypes): bool_, int8, int16, int32, int64, uint8, uint16, uint32, uint64, float16, float32, float64, complex64, complex128

Most of the array creation routines (empty, ones_like, diag, etc.)

Most of the array manipulation routines (reshape, rollaxis, concatenate, etc.)

All operators with broadcasting

All universal functions for elementwise operations (except those for complex numbers)

Linear algebra functions, including product (dot, matmul, etc.) and decomposition (cholesky, svd, etc.), accelerated by cuBLAS and cuSOLVER

Multi-dimensional fast Fourier transform (FFT), accelerated by cuFFT

Reduction along axes (sum, max, argmax, etc.)

Official website

Tutorial and documentation

Enter your contact information to continue reading