Commit 1ba98a2
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
- test_autolens/potential_correction
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
26 | 26 | | |
27 | 27 | | |
28 | 28 | | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
0 commit comments