[ummonk]

Yeah, that's definitely available in Python frameworks. (I was wrong to mention Numba - I think CuPy would be a better example: https://docs.cupy.dev/en/stable/overview.html)

[cdsousa]

That's right, I found it: https://docs.cupy.dev/en/stable/tutorial/kernel.html#kernel-...