Cupy

CuPy combines the numerical computing library NumPy and SciPy interfaces with graphics processing unit (GPU) acceleration.