code: def test_named_sharding_workflow(process_id):
      mesh = Mesh(jax.devices(), axis_names=("device"))
      sharding = NamedSharding(mesh, PartitionSpec("device"))
  
      run_distributed_sharing_test(
          sharding=sharding, process_id=process_id
      )
  
      log.info("Passed named sharding workflow test")
  
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 6 for calling a global function with example usage: test_named_sharding_workflow(process_id) without return types