Skip to content

build: widen jax/jaxlib cap to <0.12.0 (PyAutoFit#1510) - #152

Merged
Jammy2211 merged 1 commit into
mainfrom
feature/message-log-partition-tuple-shape
Aug 22, 2026
Merged

build: widen jax/jaxlib cap to <0.12.0 (PyAutoFit#1510)#152
Jammy2211 merged 1 commit into
mainfrom
feature/message-log-partition-tuple-shape

Conversation

@Jammy2211

Copy link
Copy Markdown
Collaborator

Widens the jax/jaxlib cap from <0.11.0 to <0.12.0 and replaces the stale "Cap stays <0.11" comment. The cap was held back by the message shape regression fixed in PyAutoLabs/PyAutoFit#1513merge that first (library-first order, tracked in PyAutoFit#1510). The Intel-macOS environment markers and the jaxnnls==1.0.1 pin are untouched.

Merge gate — satisfied

Every supported environment is Python >=3.12 and jax 0.11 requires >=3.12, so this widen puts jax 0.11 in front of all users with no fallback cohort. Note the downstream CI loop cannot pre-verify this: PyAutoGalaxy/PyAutoLens CI resolves jax from this repo's cap, so their green CI before this merge only exercises 0.10. The gate was therefore run as forced jax==0.11.1 installs against PyAutoFit#1513's branch:

suite jax 0.10.2 jax 0.11.1
test_autofit (with [optional] extras) 2030 passed 2030 passed
test_autoarray verified clean (#1510 audit) verified clean (#1510 audit)
test_autogalaxy 1103 passed, 1 skipped 1103 passed, 1 skipped
test_autolens 532 passed, 1 skipped 532 passed, 1 skipped
test_autonerves 157 passed 157 passed

Side effects of the widen (named in the cap comment)

  • jax 0.11 declares numpy>=2.1 and scipy>=1.15, so those become the effective floors for every install that resolves jax (compatible with the family's numpy<3 / scipy<=1.17.1 pins, but numpy 2.0.x users get bumped).
  • The 0.10→0.11 delta relevant to this stack was audited in PyAutoFit#1510: the broadcast_arrays/meshgrid/ogrid container changes (only broadcast_arrays bit, fixed in #1513), jnp.empty now uninitialized and the take_along_axis negative-index default flip (no live call sites in the family), and the jax.lax.dce_sink_p removal (unused).

🤖 Generated with Claude Code

https://claude.ai/code/session_01EauVX6vD9k1N4PXaEf2wvo


Generated by Claude Code

The <0.11.0 cap guarded the message shape regression fixed in PyAutoFit
(jnp.broadcast_arrays list -> tuple); with that fix on PyAutoFit main the
cap can widen. jax 0.11 requires numpy>=2.1 and scipy>=1.15, noted at the
cap since the widen raises those effective floors for every install that
resolves jax.

Merge order: PyAutoFit#1510's fix merges first, and PyAutoGalaxy /
PyAutoLens must be verified green under jax 0.11 before this merges —
their CI resolves jax from this cap, so pre-widen green CI does not cover
0.11; verify via matching-branch CI runs against this branch or forced
jax==0.11.1 installs.

Co-Authored-By: Claude <[email protected]>
Claude-Session: https://claude.ai/code/session_01EauVX6vD9k1N4PXaEf2wvo
@Jammy2211
Jammy2211 merged commit 6eae7e4 into main Aug 22, 2026
3 checks 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.

2 participants