cupy/cupy

What it solves

CuPyは、CPUでの大規模な数値計算を実行する際の性能ボトルネックをボトルネックを解消するために、PythonユーザーがNumPyおよびSciPyのコードをGPU上で実行できるようにします。既存のCPUベースの配列操作をGPU加速するために、コードをゼロから書き直す必要がありません。

How it works

CuPyは、NumPyおよびSciPyのドロップイン・リプレイスメントとして機能します。NVIDIA CUDAまたはAMD ROCmプラットフォームをターゲットとする互換性のある配列ライブラ0ライブラリを実装し、同一の API コールを GPU ハードウェア上で実行することを可能にします。高度なユーザー向けには、C/C++ 統合のための RawKernels、パフォーマンス・オプティマイゼーションのための Streams、CUDA Runtime APIs への直接アクセスを提供する低レベルの CUDA 機能へのアクセスも提供します。

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