Skip to content

fix(release-profile): jax_grad JAX-enable + no_run slow multi/delaunay_mge #167

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Jammy2211 merged 1 commit into from
Jul 14, 2026
Merged

fix(release-profile): jax_grad JAX-enable + no_run slow multi/delaunay_mge #167

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions config/build/env_vars_release.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,13 @@ overrides:
# needs JAX enabled; full-res is already the default.
- pattern: "jax_substructure/"
set: { PYAUTO_DISABLE_JAX: "0" }
# jax_grad scripts compute autodiff gradients (jax.value_and_grad) and assert
# finite, non-zero gradients — JAX MUST be enabled, or use_jax=False makes the
# likelihood numpy and the gradient of a non-traced function is all-zeros
# (AssertionError: Gradient is all zeros). autogalaxy_workspace_test already
# has this override; it was missing here (release-validation PyAutoHeart#72).
- pattern: "jax_grad/"
set: { PYAUTO_DISABLE_JAX: "0" }
# Database scrape scripts need real search output on disk with JAX enabled;
# PYAUTO_TEST_MODE/SMALL_DATASETS/FAST_PLOTS already match this profile's
# defaults, only PYAUTO_DISABLE_JAX needs flipping.
Expand Down
1 change: 1 addition & 0 deletions config/build/no_run.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,3 +32,4 @@
- jax_likelihood_functions/imaging/mge_group # NEEDS_FIX 2026-04-10 - timeout in JAX likelihood function benchmark
- jax_grad/imaging_lp # NEEDS_FIX 2026-04-10 - JAX traceback in gradient computation for light profile
- jax_grad/imaging_mge # NEEDS_FIX 2026-04-10 - AssertionError: Gradient is all zeros in MGE gradient computation
- jax_likelihood_functions/multi/delaunay_mge # SLOW 2026-07-14 - real-search JAX Delaunay-MGE multi-band likelihood exceeds the 1800s mode=release cap; speedup tracked by the Profiling Agent (PyAutoHeart#72). Not a bug.
Loading