Neural ODE works for small networks, but throws error for larger networks (getindex() method)?

ForwardDiffSensitivity wasn’t compatible with u0 as a scalar (we always used arrays!). Fixed that here:

Other adjoints will throw a nice error, but instead of fixing the error message here, this dispatch was easy to just solve.