julia> function f_test_1(A, x)
u = A*x[2:end] .+ x[1]
return u
end
f_test_1 (generic function with 1 method)
julia> J_f_1(A, x) = Enzyme.jacobian(Reverse, θ -> f_test_1(A, θ), x, Val(1))
J_f_1 (generic function with 1 method)
julia> x = ones(6)
julia> using LinearAlgebra
julia> A = Matrix{Float64}(LinearAlgebra.I, 5, 5)