CuPy: A NumPy-compatible matrix library accelerated by CUDA

https://cupy.chainer.org/

Leave a comment