fix: gate autofit[jax] to python>=3.11 in the jax extra

Follow-up to #530, from an independent Codex review of that commit.

#530 pointed autogalaxy[jax] at autofit[jax] so optax follows the jax chain.
PyAutoFit#1426 then marked autofit's own optax entry python>=3.11 after an
unmarked optax took every 3.9 CI leg red with resolution-too-deep.

But that fix lives on autofit's main. Any *released* autofit still expands
autofit[jax] to an unmarked optax — confirmed against PyPI:

    autofit 2026.7.27.1 -> 'optax>=0.2.5; extra == "jax"'   (no marker)

So autogalaxy[jax] was only safe when autofit came from a source checkout,
which is precisely why CI went green while the released chain stayed exposed.
CI installs all five libraries with -e, so it could never have caught this.

Marking the entry here makes the chain correct whichever autofit version pip
resolves, and restores exact pre-#530 behaviour below 3.11: every member of
the extra now carries the same gate, so the extra is a clean no-op there.

Verified in the built wheel metadata — both jax-extra entries evaluate
False on 3.9/3.10 and True on 3.11/3.12.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

Author: 2 people

pyproject.toml

@@ -53,7 +53,14 @@ jax = [
     # searches (MultiStartAdam / MultiStartProdigy) import. Without it the chain
     # autolens[jax] -> autogalaxy[jax] stopped at autonerves[jax], so `pip
     # install autolens[jax]` left those searches raising ImportError.
-    "autofit[jax]",
+    #
+    # Marked >= 3.11 so this entry matches every other member of the extra.
+    # autofit's own optax entry only gained that marker on main; any *released*
+    # autofit still expands autofit[jax] to an unmarked optax, and resolving
+    # that below 3.11 sends pip backtracking to `resolution-too-deep`. Gating
+    # here makes the chain correct whichever autofit version pip resolves,
+    # rather than only when autofit comes from a source checkout.
+    "autofit[jax]; python_version >= '3.11'",
     "jax_zero_contour>=2.0.0,<3.0.0; python_version >= '3.11'"
 ]
 optional = [