CuPy: 面向 GPU 的 NumPy & SciPy (github.com)

CuPy 是一个与 NumPy/SciPy 兼容的数组库，用于在 NVIDIA CUDA 或 AMD ROCm 平台上进行 Python GPU 加速计算。CuPy 可以直接替代 NumPy/SciPy，在 GPU 上运行现有代码。它还提供了对底层 CUDA 功能的访问，例如 RawKernels、Streams 和 CUDA Runtime API。