Speed of vectorized vs for-loops using Zygote

I should probably register it, there’s just one or two more things…

It is in principal possible to copy the function out of the verbose=2 output above – there is one called Make which is more or less laplace_for, and one called ∇Make which is what Zygote.@adjoint wants. But it’s a bit messy (to solve various issues).

1 Like