[jax:docs] Update symbol for profiling.

danielsuo · Google-ML-Automation · commit a2bd4d3ed77a · 2025-11-21T07:34:12.000-08:00
PiperOrigin-RevId: 835213384
diff --git a/docs/notebooks/host-offloading.md b/docs/notebooks/host-offloading.md
@@ -637,4 +637,4 @@ Note: The optimizer states can be compared for numerical equivalence using `jax.
 
 ## Tools for Host Offloading
 
-{func}`jax.stages.Compiled.memory_analysis` API is utilized above to get memory usage information. For device memory analysis, refer to :doc:`device_memory_profiling`. The profiling tools described in {ref}`profiling` can help measure memory savings and performance impact from host offloading.
+{func}`jax.stages.Compiled.memory_analysis` API is utilized above to get memory usage information. For device memory analysis, refer to {doc}`../device_memory_profiling`. The profiling tools described in {doc}`../profiling` can help measure memory savings and performance impact from host offloading.

Original file line number	Diff line number	Diff line change
@@ -637,4 +637,4 @@ Note: The optimizer states can be compared for numerical equivalence using `jax.
`637`	`637`
`638`	`638`	`## Tools for Host Offloading`
`639`	`639`
`640`		-{func}`jax.stages.Compiled.memory_analysis` API is utilized above to get memory usage information. For device memory analysis, refer to :doc:`device_memory_profiling`. The profiling tools described in {ref}`profiling` can help measure memory savings and performance impact from host offloading.
	`640`	+{func}`jax.stages.Compiled.memory_analysis` API is utilized above to get memory usage information. For device memory analysis, refer to {doc}`../device_memory_profiling`. The profiling tools described in {doc}`../profiling` can help measure memory savings and performance impact from host offloading.