Merge pull request #33048 from IvyZX:core_map

PiperOrigin-RevId: 840363849

Author: Google-ML-Automation

docs/conf.py

@@ -143,6 +143,7 @@ def _do_not_evaluate_in_jax(
     'pallas/tpu/distributed.md',
     'pallas/tpu/sparse.md',
     'pallas/tpu/matmul.md',
+    'pallas/tpu/core_map.md',
     'jep/9407-type-promotion.md',
     'autodidax.md',
     'autodidax2_part1.md',
@@ -245,6 +246,7 @@ def _do_not_evaluate_in_jax(
     'pallas/tpu/distributed.*',
     'pallas/tpu/sparse.*',
     'pallas/tpu/matmul.*',
+    'pallas/tpu/core_map.*',
     'distributed_data_loading.*',
     'notebooks/host-offloading.*',
 ]