Closest thing to mind is Numba's cuda JIT compilation : https://numba.pydata.org/numba-doc/latest/cuda/index.html https://numba.pydata.org/numba-doc/latest/cuda

Closest thing to mind is Numba's cuda JIT compilation : https://numba.pydata.org/numba-doc/latest/cuda/index.html https://numba.pydata.org/numba-doc/latest/cuda/index.html Then you have Cudapy : https://github.com/oulgen/CudaPy https://github.com/oulgen/CudaPy But in my opinion, the most future proof solutions are higher level frameworks like Numpy, Jax and Tensorflow. TensorFlow and Jax can JIT compile Python functions to GPU (tf.function).