If you don&#x27;t mind trying yet another library, care to have a look at arrayfire-python ? <a href="https:&#x2F;&#x2F;github.com&#x2F;arrayfire&#x2F;arrayfire-python" rel="nofollow">https:&#x2F;&#x2F;github.com&#x2F;arrayfire&#x2F;arrayfire-python</a><p>We started off as a C&#x2F;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.