array-library

CuPy是一个GPU加速的库，可作为NumPy/SciPy的直接替代品，支持在NVIDIA CUDA和AMD ROCm平台上进行高效的数组运算。