How to achieve good performance with Zygote.pushforward on a neural network

[de-souza]

Hi,

I have the same problem, and I haven’t found a way to solve it. I answered your question on the new topic you created:

Speed up compilation with Zygote and PINN

Hi, Based on what was said in a recent topic, I would recommend the approach taken by TaylorDiff.jl for higher order automatic differentiation. See the discussion and links in that topic: https://discourse.julialang.org/t/is-it-possible-to-do-nested-ad-elegantly-in-julia-pinns/98888