Add host offloading docs to public website

yashk2810 · Google-ML-Automation · commit 324280549b41 · 2025-05-08T17:25:33.000-07:00
PiperOrigin-RevId: 756522858
diff --git 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
@@ -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.

Original file line number	Diff line number	Diff line change
`@@ -143,7 +143,6 @@ def _do_not_evaluate_in_jax(`
`143`	`143`	`'autodidax2_part1.md',`
`144`	`144`	`'sharded-computation.md',`
`145`	`145`	`'ffi.ipynb',`
`146`		`- 'notebooks/host-offloading.ipynb',`
`147`	`146`	`]`
`148`	`147`
`149`	`148`	`# The name of the Pygments (syntax highlighting) style to use.`