cupy

CuPy 是一个兼容 NumPy 和 SciPy 的 GPU 加速数组库，专为 Python 编程语言设计。它能够无缝替代现有的 NumPy/SciPy 代码，在 NVIDIA CUDA 或 AMD ROCm 平台上运行，从而显著提高计算密集型任务的执行效率。除了提供与 NumPy 相似的高级功能外，CuPy 还支持直接访问低级别的 CUDA 特性，如通过 RawKernels 传递 `ndarray` 给 CUDA C/C++ 程序、使用 Streams 优化性能以及调用 CUDA Runtime API。这使得 CuPy 不仅适用于科学计算和数据分析领域内的常规应用，还非常适合需要高性能并行处理能力的深度学习、图像处理等场景。