Hi
I have the following snippet
329         for i in range(n):
330             foo[i, :] = functor(x - i)
331             bar[i, :] = functor(y - i)

Clearly these operations are parallelizable and I’m looking into something like numpy apply_along_axis or any operation that would make my stuff parallel.
Functor uses PyTorch operations.
Any ideas?