Update jax.host_id() and jax.host_count() to jax.process_index() and jax.process_count(), respectively.#378
Open
copybara-service[bot] wants to merge 1 commit intomainfrom
Open
Update jax.host_id() and jax.host_count() to jax.process_index() and jax.process_count(), respectively.#378copybara-service[bot] wants to merge 1 commit intomainfrom
copybara-service[bot] wants to merge 1 commit intomainfrom
