O4d: JAX correctness, bounded multipeak, and GPU precompute

.github/workflows/ci.yml

@@ -534,6 +534,16 @@ jobs:
               except Exception:
                   print('%-10s (absent)' % n)
           "
+      - name: Run GPU precompute CPU regression gate
+        if: matrix.shard == 1
+        run: bash .travis/test-gpu-precompute.sh
+      - name: Run optional waveform conditioning CPU tests
+        if: matrix.shard == 1
+        env:
+          JAX_ENABLE_X64: "1"
+          JAX_PLATFORM_NAME: cpu
+          JAX_PLATFORMS: cpu
+        run: python -m pytest -q MonteCarloMarginalizeCode/Code/test/waveforms/test_gpu_waveform.py
       - name: Run jax_ile CPU regression gate
         env:
           JAX_PLATFORMS: cpu
@@ -778,6 +788,7 @@ jobs:
             MonteCarloMarginalizeCode/Code/test/test_seq_warmstart_seed.py \
             MonteCarloMarginalizeCode/Code/test/test_fairdraw_double_weighting.py \
             MonteCarloMarginalizeCode/Code/test/test_portfolio_fairdraw_backend.py \
+            MonteCarloMarginalizeCode/Code/test/test_gmm_truncated_score.py \
             MonteCarloMarginalizeCode/Code/test/test_rvs_record.py
       - name: Audit _rvs consumers against the fair-draw rebind
         # sampler._rvs is rebound to an EXPORT resample at the end of integrate_log, and five
@@ -810,7 +821,9 @@ jobs:
         # --use-jax-ile and --ile-exe land in ILE.sub/ILE_puff.sub/ILE_extr.sub,
         # and --use-jax-ile is refused at DAG-build time with --calmarg-envelope-directory.
         run: |
-          python -m pytest -q MonteCarloMarginalizeCode/Code/test/test_jax_ile_selectable.py
+          python -m pytest -q \
+            MonteCarloMarginalizeCode/Code/test/test_jax_ile_selectable.py \
+            MonteCarloMarginalizeCode/Code/test/test_jax_fairdraw_postprocess.py
       - name: Run q-time-pregrid DAG-build test (--internal-ile-q-time-pregrid-factor)
         # Full subprocess DAG build against the same reference ini/coinc, same reason as
         # the step above (PR #281 follow-up review, MAJOR #1): asserts

.travis/test-core-units.sh

@@ -60,6 +60,8 @@ FILES=(
   # -- likelihood dispatch
   "$C/RIFT/likelihood/test_td_dispatch_epoch.py"
   "$C/RIFT/likelihood/test_precompute_crossterm_batching.py"
+  "$C/test/test_jax_template_finalization.py"
+  "$C/test/test_response_order.py"
   "$C/test/test_ile_scalar_edge_cases.py"
   "$C/test/test_mcsamplerGPU_cdf_inverse_scalar_probe.py"
   "$C/test/test_srate_resample_time_marginalization.py"
@@ -159,6 +161,8 @@ done
 #            found it passing 15/15 on a runner with htcondor absent)
 #   358/346  + test_mcsamplerGPU_cdf_inverse_scalar_probe.py (11 tests: mcsamplerGPU.cdf_inverse
 #            fed odeint's float probe to len(x) pdfs; the t_ref wiring in all three ILE drivers)
+#   378/366  + test_response_order.py (8 tests: SNR tightening, Halton independence,
+#            compound-axis semantics, reference resolution/tail charging, and bank preflight)
 #
 # RAISE these when files are added: a floor left at the old value passes while covering less,
 # which is the failure this gate exists to catch.
@@ -173,12 +177,13 @@ done
 # direction that matters: 350 >= 347 passes today, and if pytest-subtests ever leaves the
 # runner's closure the count falls back to 347 and still passes.  Pinning 350 would turn an
 # unrelated dependency change into a red gate.
-EXPECTED_TESTS=370
+# Two XML/grid template-finalization regressions, with no added skips.
+EXPECTED_TESTS=380
 # Outcomes, not just exit status: a collection floor cannot see a test that collects, runs and
 # asserts nothing, and a pytest.skip can quietly absorb a lost gate.  The 12 skips are
 # environment legs -- cupy in test_seeding_reproducibility, device legs in
 # test_dslice_device_native, and the xfail in test_uv_symmetry.
-EXPECTED_PASSED=358
+EXPECTED_PASSED=368
 MAX_SKIPPED=12
 
 # The floors must be INTEGERS, and this is checked rather than assumed.  `[ 347 -lt FOO ]` does

.travis/test-gpu-precompute.sh

@@ -0,0 +1,41 @@
+#!/usr/bin/env bash
+# CPU coverage of the optional GPU path: run each file in a fresh process so
+# synthetic import stubs and directory-local conftest modules cannot leak.
+set -euo pipefail
+cd "$(dirname "$0")/.."
+PYTHON_BIN="${RIFT_JAX_PYTHON:-${PYTHON:-python}}"
+export PYTHONPATH="$PWD/MonteCarloMarginalizeCode/Code${PYTHONPATH:+:$PYTHONPATH}"
+export JAX_PLATFORM_NAME=cpu JAX_PLATFORMS=cpu JAX_ENABLE_X64=1
+export OMP_NUM_THREADS="${OMP_NUM_THREADS:-1}"
+C="MonteCarloMarginalizeCode/Code/test"
+FILES=(
+  "$C/gpu_precompute/test_array_contract.py"
+  "$C/gpu_precompute/test_basis_reuse.py"
+  "$C/gpu_precompute/test_classic_compound_waveform_forwarding.py"
+  "$C/gpu_precompute/test_cross_driver_parity.py"
+  "$C/gpu_precompute/test_device_handoff_adversarial.py"
+  "$C/gpu_precompute/test_dispatch_contract.py"
+  "$C/gpu_precompute/test_highlevel_integration.py"
+  "$C/gpu_precompute/test_lal_fft_oracle.py"
+  "$C/gpu_precompute/test_smoke_config_contract.py"
+  "$C/gpu_precompute/test_streamed_v_weighting.py"
+  "$C/test_gpu_jax_handoff.py"
+  "$C/waveforms/test_gpu_legacy_compat.py"
+  "$C/waveforms/test_gpu_waveform.py"
+)
+report_dir="$(mktemp -d)"
+for file in "${FILES[@]}"; do
+  report="$report_dir/$(basename "$file").xml"
+  "${PYTHON_BIN}" -m pytest -q -p no:cacheprovider --junit-xml="$report" "$file"
+  # CUDA and optional waveform backends may skip; every file must still run
+  # at least one CPU assertion. Pytest success alone also permits all-skipped.
+  "${PYTHON_BIN}" - "$report" <<'PY'
+import sys
+import xml.etree.ElementTree as ET
+cases = list(ET.parse(sys.argv[1]).iter("testcase"))
+passed = sum(not any(case.find(tag) is not None
+                     for tag in ("skipped", "failure", "error")) for case in cases)
+if passed < 1:
+    raise SystemExit("GPU precompute CPU gate ran no passing tests: " + sys.argv[1])
+PY
+done

.travis/test-jax.sh

@@ -43,6 +43,7 @@ fi
   || { echo "test-jax.sh: numpyro unavailable (needed by test_nuts_phimarg)" >&2; exit 1; }
 
 export JAX_PLATFORMS="${JAX_PLATFORMS:-cpu}"
+export JAX_ENABLE_X64="${JAX_ENABLE_X64:-1}"
 export OMP_NUM_THREADS="${OMP_NUM_THREADS:-1}"
 
 JAXDIR="MonteCarloMarginalizeCode/Code/test/jax"
@@ -93,6 +94,9 @@ JAXDIR="MonteCarloMarginalizeCode/Code/test/jax"
 #                                         result write order) because the defects
 #                                         they pin live at call sites, where a
 #                                         helper-level assertion cannot see them.
+#   test_smc_evidence.py               2  the SMC evidence product averages over
+#                                         every walker, including zero-likelihood
+#                                         walkers, and keeps the all-finite case.
 #   test_jax_tempering_chooser.py     45  the --adapt-weight-exponent chooser and the
 #                                         tempering-cost law
 #                                         ESS/N = [beta(2-beta)]^(dim/2) it rests on.
@@ -509,6 +513,7 @@ FILES=(
   "${JAXDIR}/test_jax_time_quadrature.py"
   "${JAXDIR}/test_jax_terminal_time_marginalization.py"
   "${JAXDIR}/test_jax_likelihood.py"
+  "${JAXDIR}/test_jax_banded_data_term.py"
   "${JAXDIR}/test_jax_endtoend.py"
   "${JAXDIR}/test_jax_slowrot_coeffs.py"
   "${JAXDIR}/test_jax_slowrot_wrapper.py"
@@ -518,6 +523,7 @@ FILES=(
   "${JAXDIR}/test_nuts_phimarg.py"
   "${JAXDIR}/test_jax_av.py"
   "${JAXDIR}/test_jax_fairdraw_export.py"
+  "${JAXDIR}/test_smc_evidence.py"
   "${JAXDIR}/test_jax_tempering_chooser.py"
   "${JAXDIR}/test_tvals_grid_convention.py"
   "${JAXDIR}/test_interp_choices.py"
@@ -535,6 +541,7 @@ FILES=(
   "${JAXDIR}/test_joint_anglemarg_peaklocal.py"
   "${JAXDIR}/test_angle_marg_peaklocal_wiring.py"
   "${JAXDIR}/test_angle_marg_multipeak_wiring.py"
+  "${JAXDIR}/test_angle_marg_multipeak_jax.py"
   "${JAXDIR}/test_limit_distance_jax.py"
   "${JAXDIR}/test_direct_marginalization_planner.py"
   "${JAXDIR}/test_time_first_peaklocal.py"
@@ -1067,7 +1074,14 @@ fi
 # three EXPECTED_TESTS= assignments (761, 755, 762; last wins); this is the single one.
 # 2026-09-10: +22 value-only AV/portfolio, prior-window, wrapper, and driver
 # contract tests in test_jax_av.py.
-EXPECTED_TESTS=785
+# The multipeak host-side fix (#317) adds two more tests in
+# test_angle_marg_multipeak_wiring.py, for 787 tests total.
+# The bounded multipeak suite has 53 tests, including real acceptance/AD,
+# CLI configuration, invalid guards, and explicit drop/refuse publication.
+# 787 + 53 = 840.
+# 2026-09-12: +15 compact banded-data contraction value, AD, tile/padding,
+# scratch-budget, empty-batch, and graph-size tests. 840 + 15 = 855.
+EXPECTED_TESTS=857
 
 echo "== collection floor check (expect >= ${EXPECTED_TESTS} tests) =="
 collect_out="$("${PYTHON_BIN}" -m pytest --collect-only -q -p no:cacheprovider "${DESELECT[@]}" "${FILES[@]}" 2>&1)"

CHANGES.rst

@@ -1,6 +1,13 @@
 0.0.18.0
 ------------
 development tree is rift_O4d; PRs refer to oshaughn/research-projects-RIT.
+
+** BUG FIX, jax ILE AV/portfolio: production pseudo-pipe controls for the
+   distance prior, internal sample rate, mode retention, detector cutoffs, PSD
+   conditioning, and time interpolation are now honored by the JAX driver.
+   Previously accepted compatibility options could silently change a real-data
+   likelihood.  The AV distance prior now supports Euclidean/volumetric and
+   ``pseudo_cosmo`` and refuses unsupported choices before construction.
    - (rc0) O4d base refresh: modern Python/numpy support, portable GPU execution, generic workflow
      backends, hyperpipe and simulation-manager support; distance-likelihood export, parsimonious
      placement preview, waveform utilities and diagnostics (PRs #129, #132, #135, #143).
@@ -24,6 +31,10 @@ development tree is rift_O4d; PRs refer to oshaughn/research-projects-RIT.
      publication.  Extend slow-rotation/finite-size response support and cross-term batching; expand
      CPU/JAX regression and CI-roster coverage (fork PRs #214, #245, #247, #255, #268, #270,
      #274, #280--#285, #294, #301--#315, #319).
+     The terminal pseudo-pipe stage now recognizes JAX-ILE's tabular fair-draw
+     sidecars and joins them to the paired intrinsic likelihood records.  It no
+     longer sends JAX output through the XML-only converter, which could exit
+     successfully while producing a header-only posterior.
 
 0.0.17.12
 ---------