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[bot] merged 1 commit intomasterfrom Sep 26, 2025
Merged
[trax] Explicitly set jax_pmap_shmap_merge=False.#1814copybara-service[bot] merged 1 commit intomasterfrom
jax_pmap_shmap_merge=False.#1814copybara-service[bot] merged 1 commit intomasterfrom
Commits
Commits on Sep 26, 2025
- authored andcommitted
