Skip to content

Pass Hessian to trust-region SciPy methods#649

Open
Whning0513 wants to merge 1 commit into
google:mainfrom
Whning0513:pass-hessian-to-trust-region-solvers-240
Open

Pass Hessian to trust-region SciPy methods#649
Whning0513 wants to merge 1 commit into
google:mainfrom
Whning0513:pass-hessian-to-trust-region-solvers-240

Conversation

@Whning0513

Copy link
Copy Markdown

Summary

  • pass a Hessian callable to SciPy for trust-region/Newton-CG methods that require it
  • compute the Hessian from the wrapped JAX objective and flatten it back into SciPy's expected array shape
  • add an isolated regression test that verifies trust-ncg receives hess

Fixes #240.

Testing

  • python -m pytest -q tests/scipy_wrappers_hessian_test.py
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

1 participant