Skip to content

Adjoint improvement#46

Merged
thibmonsel merged 6 commits intomasterfrom
adjoint_improvement
Apr 7, 2025
Merged

Adjoint improvement#46
thibmonsel merged 6 commits intomasterfrom
adjoint_improvement

Conversation

@thibmonsel
Copy link
Owner

@thibmonsel thibmonsel commented Apr 7, 2025

removed has_aux argument that was used to compute adjoints for DDE and ODEs.
Changed vector fieldargs to func_args. Also modified worflow

... instead of doing an euler schema, now the adjoint dynamics and the
dL/dtheta get integrated with the solver used during the fwd pass. Might
need to changed the hacky way of making it work with if len(ts) == 1 and
ts == t1.
... dynamics integrated seamlessly with _integrate_ode. Small doc fixes
and also some to(device) fixes for solvers.
@thibmonsel thibmonsel merged commit dc88bbd into master Apr 7, 2025
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant