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?