cupy

CuPy 是一个与 NumPy 和 SciPy 接口兼容的 GPU 加速数组计算库，专为在 NVIDIA CUDA 和 AMD ROCm 平台上高效执行科学计算而设计。它提供 GPU 版本的 ndarray、ufunc、线性代数、傅里叶变换、稀疏矩阵及随机数生成等核心功能，并支持 RawKernel、CUDA Stream、cuBLAS/cuSOLVER/cuDNN 等底层 GPU 原语调用。适用于深度学习训练/推理、大规模数值模拟、高性能图像处理等需密集并行计算的场景。