PR #13258: Give GPU compiler class access to PJRT key-value store.
PR #13258: Give GPU compiler class access to PJRT key-value store.

Imported from GitHub PR https://github.com/openxla/xla/pull/13258


Copybara import of the project:

--
735291df0028b9d3a1a73a7db8fe5104910b4ca5 by Ilia Sergachev <isergachev@nvidia.com>:

Give GPU compiler class access to PJRT key-value store.

Merging this change closes #13258

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/13258 from openxla:kv 735291df0028b9d3a1a73a7db8fe5104910b4ca5
