Rocm

cupy

NumPy/SciPy に GPU アクセラレーションエンジンを搭載、コードを変更せずに瞬時に高速化。

CuPy は GPU 向けに設計された NumPy/SciPy の代替ライブラリであり、既存の Python 配列コードを直接 NVIDIA CUDA または AMD ROCm 上で実行できるようにします。