PyOpenCL

"PyOpenCL lets you access the OpenCL parallel computation API from Python.  It tries to offer computing goodness in the spirit of its sister project PyCUDA:

• Object cleanup tied to lifetime of objects. This idiom, often called RAII in C++, makes it much easier to write correct, leak- and crash-free code.
• Completeness. PyOpenCL puts the full power of OpenCL's API at your disposal, if you wish. Every obscure get_info() query and all CL calls are accessible.
• Automatic Error Checking. All CL errors are automatically translated into Python exceptions.
• Speed. PyOpenCL's base layer is written in C++, so all the niceties above are virtually free.
• Helpful and complete Documentation as well as a Wiki.
• Liberal license. PyOpenCL is open-source under the MIT license and free for commercial, academic, and private use.
• Broad support. PyOpenCL was tested and works with Apple's, AMD's, and Nvidia's CL implementations.

To use PyOpenCL, you just need numpy and an OpenCL implementation. (See this howto for how to get one.)"

https://mathema.tician.de/software/pyopencl/

https://documen.tician.de/pyopencl/

https://mathema.tician.de/software/

https://github.com/inducer/pyopencl-feedstock