code: def run_multiprocess_workflow(process_id=0):
      jax.distributed.initialize(
          coordinator_address="localhost:12321",
          num_processes=2,
          process_id=process_id,
      )
  
      log.basicConfig(
          level=log.INFO, format=f"PID {process_id}: %(message)s"
      )
  
      print_devices(process_id=process_id)
  
      test_lax_workflow(process_id=process_id)
      test_positional_sharding_workflow(process_id=process_id)
      test_named_sharding_workflow(process_id=process_id)
  
file_path: DALI/dali/test/python/jax_plugin/jax_server.py
imports: import jax
  import jax.numpy
  import numpy
  import logging
  from test_integration import get_dali_tensor_gpu
  from jax.sharding import PositionalSharding, NamedSharding, PartitionSpec, Mesh
  import nvidia.dali.plugin.jax
label_desc: Use image node_id 7 for calling a global function with example usage: run_multiprocess_workflow(process_id) without return types