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).