Hard to beat Numba / JAX loop for generating Mandelbrot

General Usage Performance
65

I’m able to use KernalAbstractions.jl to solve this problem in “jax style”, where the same kernel can be used for CPU and GPU (CUDA, ROCM, oneAPI) backend:

4 Likes