[pmap] Remove any wrappings that have auxiliary values.#34523

copybara-service[bot] merged 1 commit intomainjax-ml/jax:mainfrom

test_858908856jax-ml/jax:test_858908856Copy head branch name to clipboard