From 56bbaa36816185da6934ba7b292c3c6c9bfb6029 Mon Sep 17 00:00:00 2001
From: Yash Katariya <yashkatariya@google.com>
Date: Thu, 8 May 2025 17:57:56 -0700
Subject: [PATCH] Add host offloading docs to public website

PiperOrigin-RevId: 756532133
---
 docs/advanced_guide.rst | 1 +
 docs/conf.py            | 1 -
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/advanced_guide.rst b/docs/advanced_guide.rst
index db2e83ae2720..1cc48b8959dd 100644
--- a/docs/advanced_guide.rst
+++ b/docs/advanced_guide.rst
@@ -14,6 +14,7 @@ operations.
    notebooks/Distributed_arrays_and_automatic_parallelization
    notebooks/explicit-sharding
    notebooks/shard_map
+   notebooks/host-offloading
    multi_process
    distributed_data_loading
 
diff --git a/docs/conf.py b/docs/conf.py
index addf0cf50676..a7a52c9db38c 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -143,7 +143,6 @@ def _do_not_evaluate_in_jax(
     'autodidax2_part1.md',
     'sharded-computation.md',
     'ffi.ipynb',
-    'notebooks/host-offloading.ipynb',
 ]
 
 # The name of the Pygments (syntax highlighting) style to use.