[pmap] Remove jax.config.pmap_shmap_merge.
#438
+29
−32
Merged