在瑞士使用CuPy的公司

It is an open-source matrix library accelerated with NVIDIA CUDA. CuPy provides GPU accelerated computing with Python. It uses CUDA-related libraries including cuBLAS, cuDNN, cuRand, cuSolver, cuSPARSE, cuFFT and NCCL to make full use of the GPU architecture.