Improving the speed for the forward solve of a Universal Differential Equation (UDE)

@ChrisRackauckas any update on the sciml side?

Even without custom rules in, everything should be possible to compile, if the right traced if/fors are in