CuPy
NumPy互換のGPU配列ライブラリ - NVIDIA CUDA対応
無料無料・オープンソース（MIT License）
プロダクト詳細
CuPy とは
Preferred Networks が開発した、NVIDIA CUDAで高速化されたNumPy互換の配列ライブラリです。
主な機能
- NumPy互換API: 既存コードを最小限の変更でGPU化
- CUDA最適化: cuBLAS, cuDNN, cuFFT, cuSPARSE 統合
- カスタムカーネル: Python内でCUDAカーネル記述可能
- AMD ROCm対応: NVIDIA以外のGPUもサポート
- 分散処理: 複数GPUでの並列処理
パフォーマンス
NumPyと比較して100倍以上の高速化も可能
対象ユーザー
- 機械学習エンジニア
- 科学計算・シミュレーション
- 大規模データ処理
