diff --git a/src/base_icnf.jl b/src/base_icnf.jl index 347c74f3..5cb4d848 100644 --- a/src/base_icnf.jl +++ b/src/base_icnf.jl @@ -24,7 +24,9 @@ function construct( reltol = sqrt(eps(one(data_type))), abstol = eps(one(data_type)), maxiters = typemax(Int), - alg = OrdinaryDiffEqDefault.DefaultODEAlgorithm(), + sensealg = SciMLSensitivity.InterpolatingAdjoint(; + autojacvec = SciMLSensitivity.ZygoteVJP(), + ), ), rng::Random.AbstractRNG = MLDataDevices.default_device_rng(device), λ₁::AbstractFloat = if aicnf <: Union{RNODE, CondRNODE}