docs: JAX installs by default — retire [jax]-extra install prose (PyAutoLens#702)

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

Author: 2 people

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

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