CuPy

CuPy is NumPy-compatible matrix library accelerated by CUDA