Skip to content

[JAX] Add an import of jax.experimental.shard_map to modules that use it. - #926

Merged
copybara-service[bot] merged 1 commit into
masterfrom
test_971871217
Aug 28, 2026
Merged

[JAX] Add an import of jax.experimental.shard_map to modules that use it.#926
copybara-service[bot] merged 1 commit into
masterfrom
test_971871217

[JAX] Add an import of jax.experimental.shard_map to modules that use…

ee63703
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar