flax
[pmap] Remove `jax.config.pmap_shmap_merge`.
#5289
Merged

[pmap] Remove `jax.config.pmap_shmap_merge`. #5289

copybara-service merged 1 commit into main from test_875565036
copybara-service
copybara-service copybara-service force pushed from 1cf41386 to 55ae8b05 106 days ago
danielsuo No public description
5486a636
copybara-service copybara-service force pushed from 55ae8b05 to 5486a636 106 days ago
copybara-service copybara-service merged 5486a636 into main 106 days ago
copybara-service copybara-service deleted the test_875565036 branch 106 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone