From 8053f7c4e8484419844da535e19eb2a6f0eb4c04 Mon Sep 17 00:00:00 2001 From: Jammy2211 Date: Wed, 19 Aug 2026 14:42:33 -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 --- .../chapter_2_lens_modeling/tutorial_8_need_for_speed.ipynb | 4 ++-- scripts/chapter_2_lens_modeling/tutorial_8_need_for_speed.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/notebooks/chapter_2_lens_modeling/tutorial_8_need_for_speed.ipynb b/notebooks/chapter_2_lens_modeling/tutorial_8_need_for_speed.ipynb index d4608fa..50c5ce2 100644 --- a/notebooks/chapter_2_lens_modeling/tutorial_8_need_for_speed.ipynb +++ b/notebooks/chapter_2_lens_modeling/tutorial_8_need_for_speed.ipynb @@ -101,7 +101,7 @@ "of an order of magnitude or more are common, and they grow with the size of the dataset and model. If you have\n", "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 autolens[jax]`), every\n", + "You do not need to do anything to switch this on. JAX installs with `autolens` 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", "`al.AnalysisImaging(dataset=dataset, use_jax=False)` (or by setting the environment variable\n", @@ -209,7 +209,7 @@ "metadata": {}, "source": [ "\n", - "from autolens import setup_notebook; setup_notebook()" + "from autolens import setup_notebook; setup_notebook()\n" ], "outputs": [], "execution_count": null diff --git a/scripts/chapter_2_lens_modeling/tutorial_8_need_for_speed.py b/scripts/chapter_2_lens_modeling/tutorial_8_need_for_speed.py index 1a79fd6..cd65a14 100755 --- a/scripts/chapter_2_lens_modeling/tutorial_8_need_for_speed.py +++ b/scripts/chapter_2_lens_modeling/tutorial_8_need_for_speed.py @@ -96,7 +96,7 @@ 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 autolens[jax]`), every +You do not need to do anything to switch this on. JAX installs with `autolens` 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 `al.AnalysisImaging(dataset=dataset, use_jax=False)` (or by setting the environment variable