From fbf34e421ced96534a0d83c4b078aef237155ce4 Mon Sep 17 00:00:00 2001 From: Jammy2211 Date: Wed, 19 Aug 2026 14:42:35 -0400 Subject: [PATCH] =?UTF-8?q?docs:=20JAX=20installs=20by=20default=20?= =?UTF-8?q?=E2=80=94=20retire=20[jax]-extra=20install=20prose=20(PyAutoLen?= =?UTF-8?q?s#702)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Fable 5 --- notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb | 4 ++-- scripts/chapter_2_modeling/tutorial_8_need_for_speed.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb b/notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb index 39008ad..df91eea 100644 --- a/notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb +++ b/notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb @@ -102,7 +102,7 @@ "over CPU of an order of magnitude or more are common, and they grow with the size of the dataset and model. If you\n", "have access to a GPU (even a modest laptop GPU), it is the single biggest speed-up available to you.\n", "\n", - "You do not need to do anything to switch this on. If JAX is installed (`pip install autogalaxy[jax]`), every\n", + "You do not need to do anything to switch this on. JAX installs with `autogalaxy` by default, and every\n", "`AnalysisImaging` object we have created in this chapter defaults to `use_jax=True` and the search compiles and\n", "batches the likelihood function automatically. You can force the plain NumPy path with\n", "`ag.AnalysisImaging(dataset=dataset, use_jax=False)` (or by setting the environment variable\n", @@ -210,7 +210,7 @@ "metadata": {}, "source": [ "\n", - "from autogalaxy import setup_notebook; setup_notebook()" + "from autogalaxy import setup_notebook; setup_notebook()\n" ], "outputs": [], "execution_count": null diff --git a/scripts/chapter_2_modeling/tutorial_8_need_for_speed.py b/scripts/chapter_2_modeling/tutorial_8_need_for_speed.py index 753f8f4..9b3a083 100755 --- a/scripts/chapter_2_modeling/tutorial_8_need_for_speed.py +++ b/scripts/chapter_2_modeling/tutorial_8_need_for_speed.py @@ -97,7 +97,7 @@ over CPU of an order of magnitude or more are common, and they grow with the size of the dataset and model. If you have access to a GPU (even a modest laptop GPU), it is the single biggest speed-up available to you. -You do not need to do anything to switch this on. If JAX is installed (`pip install autogalaxy[jax]`), every +You do not need to do anything to switch this on. JAX installs with `autogalaxy` by default, and every `AnalysisImaging` object we have created in this chapter defaults to `use_jax=True` and the search compiles and batches the likelihood function automatically. You can force the plain NumPy path with `ag.AnalysisImaging(dataset=dataset, use_jax=False)` (or by setting the environment variable