feat: update to new DI.jl

Author: avik-pal

Project.toml

@@ -1,7 +1,7 @@
 name = "SimpleNonlinearSolve"
 uuid = "727e6d20-b764-4bd8-a329-72de5adea6c7"
 authors = ["SciML"]
-version = "1.12.2"
+version = "1.12.3"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
@@ -43,7 +43,7 @@ ChainRulesCore = "1.23"
 ConcreteStructs = "0.2.3"
 DiffEqBase = "6.149"
 DiffResults = "1.1"
-DifferentiationInterface = "0.5"
+DifferentiationInterface = "0.6"
 ExplicitImports = "1.5.0"
 FastClosures = "0.3.2"
 FiniteDiff = "2.23.1"

src/utils.jl

@@ -28,19 +28,19 @@ end
 
 function value_and_jacobian(
         ad, prob::AbstractNonlinearProblem, f::F, y, x, cache; J = nothing) where {F}
-    x isa Number && return DI.value_and_derivative(f, ad, x, cache)
+    x isa Number && return DI.value_and_derivative(f, cache, ad, x)
 
     if isinplace(prob)
         if cache isa HasAnalyticJacobian
             prob.f.jac(J, x, prob.p)
             f(y, x)
             return y, J
         end
-        return DI.value_and_jacobian!(f, y, J, ad, x, cache)
+        return DI.value_and_jacobian!(f, y, J, cache, ad, x)
     else
         cache isa HasAnalyticJacobian && return f(x), prob.f.jac(x, prob.p)
-        J === nothing && return DI.value_and_jacobian(f, ad, x, cache)
-        y, J = DI.value_and_jacobian!(f, J, ad, x, cache)
+        J === nothing && return DI.value_and_jacobian(f, cache, ad, x)
+        y, J = DI.value_and_jacobian!(f, J, cache, ad, x)
         return y, J
     end
 end