From 6c92e12dc2fbae18a5a81dcadbb8828e08fdee8f Mon Sep 17 00:00:00 2001 From: Nicholas Junge Date: Thu, 2 Oct 2025 10:42:41 +0200 Subject: [PATCH] docs(pallas): Fix broken link to pallas documentation section --- jax/experimental/pallas/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jax/experimental/pallas/__init__.py b/jax/experimental/pallas/__init__.py index b8db4e430ec7..a5a5d6d0909c 100644 --- a/jax/experimental/pallas/__init__.py +++ b/jax/experimental/pallas/__init__.py @@ -15,7 +15,7 @@ """Module for Pallas, a JAX extension for custom kernels. See the Pallas documentation at -https://docs.jax.dev/en/latest/pallas.html. +https://docs.jax.dev/en/latest/pallas/index.html. """ from jax._src.pallas.core import BlockDim as BlockDim