Skip to content

test: multi shared_preloads parity script (imaging sibling of datacube)#162

Merged
Jammy2211 merged 1 commit into
mainfrom
feature/multi-shared-state-phase-3
Jul 10, 2026
Merged

test: multi shared_preloads parity script (imaging sibling of datacube)#162
Jammy2211 merged 1 commit into
mainfrom
feature/multi-shared-state-phase-3

Conversation

@Jammy2211

Copy link
Copy Markdown
Collaborator

Summary

Phase 3/4 (autolens_workspace#260): the multi shared_preloads parity script — imaging sibling of datacube/shared_preloads.py, as requested (test examples comparing to examples not using the shared API).

scripts/jax_likelihood_functions/multi/shared_preloads.py:

  • identical-exposure (2× g-band) shared-vs-unshared summed log-likelihood bit-identical, numpy AND JAX (−30741.124896855516 / −30741.124896846883);
  • realistic g + r shared-mesh graph: vmap == jit(log_likelihood_function) round-trip (−51323.05150762) — the shared path threads jax.jit end-to-end;
  • registered in smoke_tests.txt (one curated addition).

Also verified against the scoped-preloads library branch (PyAutoArray/PyAutoLens follow-up pair).

Autonomy

--auto safe (gate on issue #260; Heart YELLOW set unchanged from in-session ack).

🤖 Generated with Claude Code

Identical-exposure shared-vs-unshared parity (numpy + JAX, bit-identical) + realistic g+r
shared-mesh graph vmap/jit agreement; registered in smoke_tests.txt.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@Jammy2211 Jammy2211 added the pending-release PR queued for the next release build label Jul 10, 2026
@Jammy2211 Jammy2211 merged commit 0612ee0 into main Jul 10, 2026
4 checks passed
@Jammy2211 Jammy2211 deleted the feature/multi-shared-state-phase-3 branch July 10, 2026 15:34
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

pending-release PR queued for the next release build

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant