We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bcfea98 commit cdb2b69Copy full SHA for cdb2b69
docs/conf.py
@@ -143,6 +143,7 @@ def _do_not_evaluate_in_jax(
143
'pallas/tpu/distributed.md',
144
'pallas/tpu/sparse.md',
145
'pallas/tpu/matmul.md',
146
+ 'pallas/tpu/core_map.md',
147
'jep/9407-type-promotion.md',
148
'autodidax.md',
149
'autodidax2_part1.md',
@@ -245,6 +246,7 @@ def _do_not_evaluate_in_jax(
245
246
'pallas/tpu/distributed.*',
247
'pallas/tpu/sparse.*',
248
'pallas/tpu/matmul.*',
249
+ 'pallas/tpu/core_map.*',
250
'distributed_data_loading.*',
251
'notebooks/host-offloading.*',
252
]
0 commit comments