How can I improve the computational performance of my code (ODEProblem solving)?

[chooron]

Hello, I am encountering another issue with building an ODE problem that involves the Lux.jl model. Could you please provide me with some suggestions to resolve this issue?