cupy
CuPy enables NumPy/SciPy code to run on GPU with minimal changes, providing massive speedups for array operations.
CuPy is a NumPy/SciPy-compatible array library for GPU-accelerated computing, offering drop-in replacement to run existing code on NVIDIA CUDA or AMD ROCm platforms.