code: def make_cpu_client() -> xla_client.Client:
      if xla_extension_version >= 223:
          collectives: xla_client._xla.CpuCollectives | None = None
          if _CPU_ENABLE_GLOO_COLLECTIVES.value:
              collectives = xla_client._xla.make_gloo_tcp_collectives(  # type: ignore
                  distributed_client=distributed.global_state.client,
              )
          return xla_client.make_cpu_client(  # type: ignore
              distributed_client=distributed.global_state.client,
              node_id=distributed.global_state.process_id,
              num_nodes=distributed.global_state.num_processes,
              collectives=collectives,
          )
      elif xla_extension_version >= 216:
          # TODO(phawkins): remove type: ignore after updating jaxlib version used for
          # mypy checks.
          return xla_client.make_cpu_client(  # type: ignore
              distributed_client=distributed.global_state.client,
              node_id=distributed.global_state.process_id,
              num_nodes=distributed.global_state.num_processes,
          )
      else:
          return xla_client.make_cpu_client()
  
imports: from __future__ import annotations
  from __future__ import annotations
  from collections.abc import Mapping
  import dataclasses
  from functools import lru_cache, partial
  import importlib
  import json
  import logging
  import os
  import pkgutil
  import platform
  import sys
  import threading
  from typing import Any, Callable, Union
  import warnings
  from jax._src import config
  from jax._src import distributed
  from jax._src import traceback_util
  from jax._src import util
  from jax._src import hardware_utils
  from jax._src.cloud_tpu_init import maybe_import_libtpu
  from jax._src.lib import cuda_versions
  from jax._src.lib import xla_client
  from jax._src.lib import xla_extension
  from jax._src.lib import xla_extension_version
label_desc: Use image node_id 5 for calling a global function with example usage: make_cpu_client() and returns: xla_client, xla_client, xla_client