[WIP] Improvements to the dynamic optimization interface#4394
Draft
SebastianM-C wants to merge 9 commits into
Draft
[WIP] Improvements to the dynamic optimization interface#4394SebastianM-C wants to merge 9 commits into
SebastianM-C wants to merge 9 commits into
Conversation
SebastianM-C
changed the title
FunctionWrapper parameters (e.g., interpolators) fail during fixpoint_sub with fold=Val(true) because they can't accept JuMP types. Register them as JuMP nonlinear operators instead, with automatic derivative detection via Symbolics' derivative_rule system. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Allow users to provide constraint scaling, variable bounds, and function-valued start trajectories at problem construction time. All keys are normalized via default_toterm for consistent lookup. Thread kwargs through all backend extensions (InfiniteOpt, CasADi, Pyomo). Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Replace inequality-constraint-based bounds with JuMP set_lower_bound/ set_upper_bound via dispatched set_variable_bounds!. Extract bounds logic into extract_variable_bounds with support for user-provided bounds that override metadata. This gives Ipopt better barrier method structure for optimal control problems. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Allow users to provide function-valued initial trajectories for state variables via the initial_trajectory kwarg. Functions are applied via dispatched set_initial_trajectory! after variable creation. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Formulate dynamics as (∂x - tₛ*f(x)) / scale == 0 instead of scaling each side independently. This prevents degenerate tₛ → 0 solutions and improves Ipopt convergence for problems with multi-scale state variables. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Substitute observed variables before parameters so that observed equations referencing parameters get fully resolved during fixpoint_sub. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Set print_level after set_optimizer to prevent the optimizer reset from discarding the attribute. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Extend the optimal control bounds system to support observed variables: - `extract_variable_bounds` now returns `observed_bounds` from variable metadata and user `bounds` dict (user bounds take priority) - InfiniteOpt backend lifts observed bounds into auxiliary bounded decision variables with equality constraints, which interior-point solvers handle much more efficiently than nonlinear inequalities - Auxiliary variables get proper start values computed from the initial state, and equality constraints are scaled via `scales` dict or variable metadata - Change `add_initial_constraints!` from `fix` to `@constraint` to preserve variable bounds at the initial time point Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Checklist
contributor guidelines, in particular the SciML Style Guide and
COLPRAC.
Additional context
I will be splitting this in several PRs