feat: JAX default install, [jax] alias + install docs (#702) - #703
Merged
Conversation
… new no-jax CI leg (#702) Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Collaborator
Author
|
Workspace PRs: PyAutoLabs/autolens_workspace#486, PyAutoLabs/HowToLens#71 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
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 autolensis now the JAX-enabled command; the[jax]form is documented as a still-working alias; Intel Macs (no jaxlibmacosx_x86_64wheels ≥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
[jax]alias resolvesunittest-nojax) green once the Heart PR mergesFull API Changes (for automation & release notes)
Changed Behaviour
pip install autolensnow installs JAX by default via the family chain (except Intel macOS, which falls back to NumPy-only with an import-time warning).Migration
pip install autolens[jax]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.