Unwrap AutoSpecializeCallable for Enzyme in JacobianOperator path

The previous commit only unwrapped for the concrete Jacobian path
(DI.prepare_jacobian/DI.jacobian). This extends the fix to the
JacobianOperator path used by Krylov solvers (GMRES, etc.) and
backslash with concrete_jac=false.

When Enzyme is used for JVP/VJP autodiff, create a modified problem
with the raw user function so SciMLJacobianOperators' DI.pushforward!/
DI.pullback! calls don't go through FunctionWrappers' llvmcall.

Co-Authored-By: Chris Rackauckas <accounts@chrisrackauckas.com>
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: ChrisRackauckas

lib/NonlinearSolveBase/src/jacobian.jl

@@ -71,7 +71,15 @@ function construct_jacobian_cache(
         # Standardize JVP/VJP autodiff tags to match FunctionWrapper signatures
         _jvp_ad = standardize_forwarddiff_tag(jvp_autodiff, prob)
         _vjp_ad = standardize_forwarddiff_tag(vjp_autodiff, prob)
-        JacobianOperator(prob, fu, u; jvp_autodiff = _jvp_ad, vjp_autodiff = _vjp_ad)
+        # Enzyme cannot differentiate through FunctionWrappers' llvmcall.
+        # Unwrap AutoSpecializeCallable so DI sees the raw user function.
+        _prob = if is_fw_wrapped(f.f) &&
+                (_uses_enzyme_ad(_jvp_ad) || _uses_enzyme_ad(_vjp_ad))
+            @set prob.f.f = get_raw_f(f.f)
+        else
+            prob
+        end
+        JacobianOperator(_prob, fu, u; jvp_autodiff = _jvp_ad, vjp_autodiff = _vjp_ad)
     else
         if f.jac_prototype === nothing
             # While this is technically wasteful, it gives out the type of the Jacobian