Passes kv_store to ifrt::PjRtClient.
Passes kv_store to ifrt::PjRtClient.

This is required for cross-host device transfer. Devices between different hosts need a way to communicate, and we will use kv_store for that.

Update the signature of relevant APIs and pybind to create a ifrt::PjRtClient.
