Hi!
I have a function that I want to apply to each element of an array, and return the stacked result. The function has if conditions, slicing and so on. Is there a way to parallelize the function (on GPU) over batch dimension?