cupy/cupy

这是一个利用 GPU 加速数值计算的 Python 库，与 NumPy 和 SciPy 兼容。你可以轻松地将现有的 NumPy/SciPy 代码，迁移到 NVIDIA CUDA 或 AMD ROCm 平台上运行，部分情况下速度可提升 100 倍以上。