Hi, I’m trying to write a simple PINN (e.g., Newton cooling ODE) in Julia. I’ve tried Flux + Zygote but keep hitting issues from recent API changes and slow higher‑order AD. What’s the current recommended stack and a minimal working example that actually runs? Would you go with Flux, Lux, or just use NeuralPDE.jl? Tips on stable AD for second‑order derivatives are also welcome. Thanks!
Lux + Reactant+ Enzyme is the current way I think. I don’t know if NeuralPDE did the switch yet