[pmap] Remove jax.config.pmap_shmap_merge.
#1868
| Job | Run time |
|---|---|
| 2m 40s | |
| 2m 45s | |
| 5m 25s |
jax.config.pmap_shmap_merge.
#1868
| Job | Run time |
|---|---|
| 2m 40s | |
| 2m 45s | |
| 5m 25s |