code: def initialize_preemption_sync_manager(self):
      if self.preemption_sync_manager is not None:
          raise RuntimeError(
              "Preemption sync manager should only be initialized once."
          )
      self.preemption_sync_manager = (
          xla_extension.create_preemption_sync_manager()
      )
      self.preemption_sync_manager.initialize(self.client)
  
file_path: jax/jax/_src/distributed.py
imports: from __future__ import annotations
  import atexit
  from collections.abc import Sequence
  import logging
  import os
  from typing import Any
  from jax._src import clusters
  from jax._src import config
  from jax._src import xla_bridge
  from jax._src.lib import xla_extension
label_desc: Use image node_id 3 for calling the State obj's underlying member method code with example usage: obj.initialize_preemption_sync_manager() without return types