Update jax.host_id() and jax.host_count() to jax.process_index() and jax.process_count(), respectively.#839

copybara-service[bot] merged 1 commit intomastergoogle/init2winit:masterfrom

test_863031810google/init2winit:test_863031810Copy head branch name to clipboard