Skip to content

test via zygote #463

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
prbzrg wants to merge 16 commits into from
Closed

test via zygote #463

Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
7eb3b45
test via zygote
prbzrg Feb 13, 2025
18a936c
have alg
prbzrg Feb 13, 2025
d507263
use defaultalg
prbzrg Feb 13, 2025
d7dc77a
Merge branch 'main' into test-viazygote
prbzrg Feb 13, 2025
68c1ee2
disable lux nested
prbzrg Feb 14, 2025
f7fae08
back to DI
prbzrg Feb 14, 2025
a5842e8
simpler
prbzrg Feb 14, 2025
65c3185
just sensealg
prbzrg Feb 20, 2025
b23fb4a
Merge branch 'main' into test-viazygote
prbzrg Feb 25, 2025
2156358
Merge branch 'main' into test-viazygote
prbzrg Feb 28, 2025
68cfa66
Merge branch 'main' into test-viazygote
prbzrg Mar 7, 2025
c3629c3
Merge branch 'main' into test-viazygote
prbzrg Mar 29, 2025
28d0f4d
Merge branch 'main' into test-viazygote
prbzrg Mar 29, 2025
7257f87
Merge branch 'main' into test-viazygote
prbzrg Apr 17, 2025
9cd7019
Merge branch 'main' into test-viazygote
prbzrg Apr 24, 2025
d5b981b
Merge branch 'main' into test-viazygote
prbzrg May 13, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion src/base_icnf.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}
Expand Down
Loading