I’m not sure about this but have you tried simply removing the closure to see how this affects the running time?
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Zygote dozens* of times slower than manually written function | 17 | 1885 | April 21, 2022 | |
| Optimization.jl, DataInterpolations.jl and Gradients | 4 | 833 | April 21, 2023 | |
| Zygote terribly slow. What am I doing wrong? | 12 | 1847 | January 19, 2022 | |
| DataInterpolations.jl and Zygote.jl | 18 | 710 | July 8, 2024 | |
| Zygote vs. Forward Diff with Optim | 4 | 690 | March 24, 2023 |