jax

f174b00f - Replace the uses of `PjRtClient::Compile()` with `PjRtClient::CompileAndLoad()`.

Replace the uses of `PjRtClient::Compile()` with `PjRtClient::CompileAndLoad()`. This is to prepare for updating `PjRtClient::Compile()` to return an unloaded executable [1/N] PiperOrigin-RevId: 737805623