If you don't mind trying yet another library, care to have a look at arrayfire-python ? https://github.com/arrayfire/arrayfire-python https://github.com/arrayfi

If you don't mind trying yet another library, care to have a look at arrayfire-python ? https://github.com/arrayfire/arrayfire-python https://github.com/arrayfire/arrayfire-python We started off as a C/C++ library but are constantly adding support for new languages. The one benefit of ArrayFire would be the ability to use the same codebase across CPU, CUDA and OpenCL devices.