Add programmatic profiling test with session_id support in JAX.#34556

copybara-service[bot] wants to merge 1 commit intomainopenxla/xla:mainfrom

test_837805018openxla/xla:test_837805018Copy head branch name to clipboard