CuPy – NumPy-like API accelerated with CUDA

This is the CuPy documentation.