ForwardDiffSensitivity Faster than Adjoint Methods

Reading your other post, I think you sent the flamegraph from this model Composing a Neural ODE with another Neural Network