Skip to content
This repository was archived by the owner on Oct 31, 2025. It is now read-only.

[trax] Explicitly set jax_pmap_shmap_merge=False. #1814

Merged
copybara-service merged 1 commit into from
Sep 26, 2025

[trax] Explicitly set `jax_pmap_shmap_merge=False`.

31022d6
Select commit
Loading
Failed to load commit list.
Merged

[trax] Explicitly set jax_pmap_shmap_merge=False. #1814

[trax] Explicitly set `jax_pmap_shmap_merge=False`.
31022d6
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar