Actually in julia you can write kernels with a subset of the julia language: https://cuda.juliagpu.org/stable/tutorials/introduction/#Writing-your-first-GPU-ke

Actually in julia you can write kernels with a subset of the julia language: https://cuda.juliagpu.org/stable/tutorials/introduction/#Writing-your-first-GPU-kernel https://cuda.juliagpu.org/stable/tutorials/introduction/#Wri... With KernelAbstractions.jl you can actually target CUDA and ROCm: https://juliagpu.github.io/KernelAbstractions.jl/stable/kernels/ https://juliagpu.github.io/KernelAbstractions.jl/stable/kern... For python (or rather python-like), there is also triton (and probably others): https://pytorch.org/blog/triton-kernel-compilation-stages/ https://pytorch.org/blog/triton-kernel-compilation-stages/