Skip to content

Commit 1ba98a2

Browse files
Jammy2211claude
authored andcommitted
Add xp-API dense kernels + iterative LM engine (potential correction phase 4)
Refactors the jax_ops kernels of Cao et al. 2025's potential_correction package onto the PyAuto xp API: al.pc.dense_util writes each dense kernel once (xp=np default, jax.numpy for accelerator execution) — evidence variants incl. the fixed-curvature Cholesky fast path, and the Levenberg-Marquardt kernels. Adds al.pc.IterFitDpsiSrcImaging, the iterative joint source+dpsi LM engine (re-ray-traces through the corrected lens via the phase-2 InputPotential each accepted step; optional gauge constraints via a KKT step) and its IterDpsiSrcInvAnalysis. Fixes an upstream NaN bug in split_cross_from (percentile over unbounded-cell sentinels). Phase 4 of #618; stacked on #621. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
1 parent 5c5b6c3 commit 1ba98a2

7 files changed

Lines changed: 1565 additions & 2 deletions

File tree

autolens/potential_correction/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,3 +26,6 @@
2626
from autolens.potential_correction.fit import FitDpsiSrcImaging
2727
from autolens.potential_correction.analysis import DpsiInvAnalysis
2828
from autolens.potential_correction.analysis import DpsiSrcInvAnalysis
29+
from autolens.potential_correction import dense_util
30+
from autolens.potential_correction.iterative import IterFitDpsiSrcImaging
31+
from autolens.potential_correction.iterative import IterDpsiSrcInvAnalysis

0 commit comments

Comments
 (0)