cupy/cupy

What it solves

CuPy 解决了在 CPU 上运行大规模数值计算时的性能瓶颈，它允许 Python 用户在 GPU 上执行 NumPy 和 SciPy 代码。它消除了从头开始重写现有基于 CPU 的数组代码以实现 GPU 加速的需求。

How it works

CuPy 作为 NumPy 和 SciPy 的直接替换方案。它实现了实现一个针对 NVIDIA CUDA 或 AMD ROCm 平台的兼容数组库，使得相同的 API 调用可以可以在 GPU 硬件上执行。对于高级用户，它还提供了对 CUDA 功能的低阶层级访问，包括用于 C/C++ 集成的 RawKernels、用于性能优化的 Streams，以及对 CUDA Runtime APIs 的直接访问。

Highlights

• Drop-in compatibility with NumPy and SciPy APIs.
• Supports both NVIDIA CUDA and AMD ROCm platforms.
• Provides low-level CUDA C/C++ integration via RawKernels.
• Includes integrated support for signal processing workflows (via cuSignal).