cupy

CuPy是NumPy和SciPy的GPU加速替代库，支持在NVIDIA CUDA和AMD ROCm平台上运行。