Skip to content

Handle infinite and fixed bounds in ALM inner solves - #185

Open
rogeriojorge wants to merge 1 commit into
mainfrom
fix/alm-bound-types
Open

rogeriojorge wants to merge 1 commit into
mainfrom
fix/alm-bound-types

Conversation

@rogeriojorge

Copy link
Copy Markdown
Member

The ALM inner solver's finite-interval transform produces NaNs for one-sided bounds such as (0, inf) and for fixed coordinates where lower equals upper. These bounds were accepted by the former JAXopt entry point, which remains available as an alias.

Handle unbounded and fixed coordinates explicitly and reflect one-sided coordinates at their finite endpoint. Preserve the existing finite-interval transform. Regression tests exercise actual ALM updates with mixed bound types and boundary starts.

Validation: tests/test_augmented_lagrangian.py — 28 passed, including seven new bound cases.

@gftd13 please review and test one-sided and fixed bounds in existing ALM callers, including parameters initialized exactly at a bound.

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