Deploying to gh-pages from @ a71bd2e 🚀

Author: lucascolley

generated/array_api_extra.at.html

@@ -359,7 +359,7 @@ <h1>array_api_extra.at<a class="headerlink" href="#array-api-extra-at" title="Li
 <div class="admonition seealso">
 <p class="admonition-title">See also</p>
 <dl class="simple">
-<dt><a class="reference external" href="https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ndarray.at.html#jax.numpy.ndarray.at" title="(in JAX)"><code class="xref py py-obj docutils literal notranslate"><span class="pre">jax.numpy.ndarray.at</span></code></a></dt><dd><p>Equivalent array method in JAX.</p>
+<dt><a class="reference external" href="https://docs.jax.dev/en/latest/_autosummary/jax.numpy.ndarray.at.html#jax.numpy.ndarray.at" title="(in JAX)"><code class="xref py py-obj docutils literal notranslate"><span class="pre">jax.numpy.ndarray.at</span></code></a></dt><dd><p>Equivalent array method in JAX.</p>
 </dd>
 </dl>
 </div>

generated/array_api_extra.testing.lazy_xp_function.html

@@ -317,7 +317,7 @@ <h1>array_api_extra.testing.lazy_xp_function<a class="headerlink" href="#array-a
 <dl class="simple">
 <dt><a class="reference internal" href="array_api_extra.testing.patch_lazy_xp_functions.html#array_api_extra.testing.patch_lazy_xp_functions" title="array_api_extra.testing.patch_lazy_xp_functions"><code class="xref py py-obj docutils literal notranslate"><span class="pre">patch_lazy_xp_functions</span></code></a></dt><dd><p>Companion function to call from the test or fixture.</p>
 </dd>
-<dt><a class="reference external" href="https://jax.readthedocs.io/en/latest/_autosummary/jax.jit.html#jax.jit" title="(in JAX)"><code class="xref py py-obj docutils literal notranslate"><span class="pre">jax.jit</span></code></a></dt><dd><p>JAX function to compile a function for performance.</p>
+<dt><a class="reference external" href="https://docs.jax.dev/en/latest/_autosummary/jax.jit.html#jax.jit" title="(in JAX)"><code class="xref py py-obj docutils literal notranslate"><span class="pre">jax.jit</span></code></a></dt><dd><p>JAX function to compile a function for performance.</p>
 </dd>
 </dl>
 </div>

generated/array_api_extra.testing.patch_lazy_xp_functions.html

@@ -274,7 +274,7 @@ <h1>array_api_extra.testing.patch_lazy_xp_functions<a class="headerlink" href="#
 <dd><p>Test lazy execution of functions tagged with <a class="reference internal" href="array_api_extra.testing.lazy_xp_function.html#array_api_extra.testing.lazy_xp_function" title="array_api_extra.testing.lazy_xp_function"><code class="xref py py-func docutils literal notranslate"><span class="pre">lazy_xp_function()</span></code></a>.</p>
 <p>If <code class="docutils literal notranslate"><span class="pre">xp==jax.numpy</span></code>, search for all functions which have been tagged with
 <a class="reference internal" href="array_api_extra.testing.lazy_xp_function.html#array_api_extra.testing.lazy_xp_function" title="array_api_extra.testing.lazy_xp_function"><code class="xref py py-func docutils literal notranslate"><span class="pre">lazy_xp_function()</span></code></a> in the globals of the module that defines the current test
-and wrap them with <a class="reference external" href="https://jax.readthedocs.io/en/latest/_autosummary/jax.jit.html#jax.jit" title="(in JAX)"><code class="xref py py-func docutils literal notranslate"><span class="pre">jax.jit()</span></code></a>. Unwrap them at the end of the test.</p>
+and wrap them with <a class="reference external" href="https://docs.jax.dev/en/latest/_autosummary/jax.jit.html#jax.jit" title="(in JAX)"><code class="xref py py-func docutils literal notranslate"><span class="pre">jax.jit()</span></code></a>. Unwrap them at the end of the test.</p>
 <p>If <code class="docutils literal notranslate"><span class="pre">xp==dask.array</span></code>, wrap the functions with a decorator that disables
 <code class="docutils literal notranslate"><span class="pre">compute()</span></code> and <code class="docutils literal notranslate"><span class="pre">persist()</span></code>.</p>
 <p>This function should be typically called by your library’s <cite>xp</cite> fixture that runs