[optax] Remove jax_pmap_shmap_merge=False flag in contrib/_complex_valued. #jax-fixit
#1601
test_864834674 was force-pushed and no longer has any new commits.
Pushing new commits will allow the pull request to be re-opened.