Skip to content

maintenance: validate blackjax 1.6.2 + mainline NSS smoke + NUTS API check - #143

Merged
Jammy2211 merged 1 commit into
mainfrom
feature/blackjax-16-upgrade-smoke
Aug 18, 2026
Merged

maintenance: validate blackjax 1.6.2 + mainline NSS smoke + NUTS API check#143
Jammy2211 merged 1 commit into
mainfrom
feature/blackjax-16-upgrade-smoke

Conversation

@Jammy2211

Copy link
Copy Markdown
Contributor

Closes #142. Phase 0(b) of the inference programme — the CPU environment half of critical-path item CP-2, run in a cloud session per human direction ("just getting install type stuff working"). The GPU MGE smoke half of CP-2 and the actual local-venv/RAL upgrades remain laptop work.

Results (all PASS)

  • Install: blackjax 1.6.2 + jax 0.10.2 (CPU) + released autofit 2026.7.29.1 resolve together with no conflicts (pip check clean). Two dependency facts recorded: PyAutoFit pins blackjax>=1.2.0 — floor only, no cap, and only in the optional extra — so the Gate-A floor bump is a one-liner; and blackjax is not a hard autofit dependency, so upgrades must pip install -U blackjax explicitly.
  • blackjax.nss 2D toy smoke: exercises the exact mainline API surface Phase 2's profiling-local runner will wire (native-space logprior_fn, num_delete, dlogz termination via state.integrator.logZ_live − logZ, ns.utils.finalise + log_weights simulated-volume ensemble, ns.utils.sample). logZ = −4.609 ± 0.069 vs analytic −4.605; posterior mean/σ recover truth. Bonus provenance: the installed 1.6.2 docstring states the num_inner_steps ≥ max(5, 2·d) rule and the upward logZ-bias direction verbatim — H2.1's citation is now verified-in-source.
  • af.BlackJAXNUTS on 1.6.2: runs unmodified (window_adaptation, nuts, diagnostics.effective_sample_size all intact) — toy posterior recovered, 0 divergences, acceptance 0.92, ESS_min 244. No PyAutoFit compatibility fix needed; the §7 Phase-6 items stay features, not fixes.

Contents

  • scripts/misc/searches/nss_smoke.py — rerunnable smoke (exits non-zero on FAIL, guards on blackjax < 1.6); run it in each environment after its upgrade.
  • results/notes/inference/phase_00_unblocking/RESULTS.md — the findings record.
  • PROGRAMME.md phase table: 0(b) → validated (cloud CPU), venv/RAL upgrades remaining.

ruff check / ruff format --check / build_readme.py --check all pass. CPU toy timings carry no profiling meaning (wrong tier, per the programme's own rules).

🤖 Generated with Claude Code

https://claude.ai/code/session_018dV9h5h76hZ6trGZKWJFHe


Generated by Claude Code

…142)

Phase 0(b) / the CPU environment half of CP-2, per human direction
(install/API validation only; GPU MGE smoke and local/RAL venv upgrades
remain laptop work):

- blackjax 1.6.2 + jax 0.10.2 (CPU) + released autofit 2026.7.29.1
  coexist cleanly; PyAutoFit pins blackjax>=1.2.0 (floor, no cap, in the
  'optional' extra) so the Gate-A floor bump has no cap to fight — but
  blackjax is not a hard autofit dependency, so upgrades must install it
  explicitly.
- scripts/misc/searches/nss_smoke.py: rerunnable CPU smoke for every
  environment's upgrade. blackjax.nss 2D toy exercises the full mainline
  API (native-space logprior_fn, num_delete, dlogz termination on
  integrator.logZ_live - logZ, ns.utils.finalise + log_weights ensemble):
  logZ -4.609 +/- 0.069 vs analytic -4.605, posterior recovered.
  af.BlackJAXNUTS runs on 1.6.2 unmodified (0 divergences, acc 0.92).
- results/notes/inference/phase_00_unblocking/RESULTS.md: the findings,
  incl. verified-in-source provenance for the H2.1 inner-steps/logZ-bias
  guidance; PROGRAMME phase table 0(b) -> validated (cloud CPU).

Co-Authored-By: Claude <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_018dV9h5h76hZ6trGZKWJFHe
@Jammy2211
Jammy2211 merged commit 7a7eeaf into main Aug 18, 2026
1 check passed
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.

maintenance: Phase 0(b) — blackjax ≥1.6.2 validation + mainline NSS smoke (CPU)

2 participants