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

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