Skip to content

feat: JAX default install, [jax] alias + install docs (#702) - #703

Merged
Jammy2211 merged 2 commits into
mainfrom
feature/jax-default-dependency
Aug 19, 2026
Merged

feat: JAX default install, [jax] alias + install docs (#702)#703
Jammy2211 merged 2 commits into
mainfrom
feature/jax-default-dependency

Conversation

@Jammy2211

Copy link
Copy Markdown
Collaborator

Summary

Closes the packaging half of #702: JAX is now a default dependency of the stack. autolens receives jax, optax and jax_zero_contour transitively through the autogalaxy → autofit → autonerves base-dependency chain (each promoted in its own repo's sibling PR), so the changes here are the [jax] extra becoming a declared no-op alias (#687), the floor comment recording the follow-up floor bump, and the install docs.

Docs (docs/installation/pip.md, conda.md): pip install autolens is now the JAX-enabled command; the [jax] form is documented as a still-working alias; Intel Macs (no jaxlib macosx_x86_64 wheels ≥0.7) are documented as automatically falling back to the fully supported NumPy-only path, with a loud import-time warning.

API Changes

None — packaging metadata and install docs only.
See full details below.

Test Plan

  • pyproject parses; [jax] alias resolves
  • Clean-venv install of the promoted chain pulls jax 0.11.1 + jaxnnls automatically
  • No-JAX CI leg (PyAutoHeart lib-tests.yml unittest-nojax) green once the Heart PR merges
Full API Changes (for automation & release notes)

Changed Behaviour

  • pip install autolens now installs JAX by default via the family chain (except Intel macOS, which falls back to NumPy-only with an import-time warning).

Migration

  • Before: pip install autolens[jax]
  • After: pip install autolens (the [jax] form still works — no-op alias)

Sibling PRs: PyAutoNerves, PyAutoFit, PyAutoArray, PyAutoGalaxy (packaging), PyAutoHeart (no-JAX CI leg). Follow-up after the first promoted release: bump the intra-family floors to it.

Generated by the PyAutoLabs agent workflow.

@Jammy2211 Jammy2211 added the pending-release PR queued for the next release build label Aug 19, 2026
… new no-jax CI leg (#702)

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

Copy link
Copy Markdown
Collaborator Author

@Jammy2211
Jammy2211 merged commit 6470e64 into main Aug 19, 2026
4 checks passed
@Jammy2211
Jammy2211 deleted the feature/jax-default-dependency branch August 19, 2026 19:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

pending-release PR queued for the next release build

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant