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
Commits
Commits on Sep 26, 2025
- authored andcommitted
