code: def _to_jax_array(dali_tensor: TensorGPU) -> jax.Array:
      """Converts input DALI tensor to JAX array.
  
      Args:
          dali_tensor (TensorGPU): DALI GPU tensor to be converted to JAX array.
  
      Note:
          This function performs deep copy of the underlying data. That will change in
          future releases.
  
      Warning:
          As private this API may change without notice.
  
      Returns:
          jax.Array: JAX array with the same values and backing device as
          input DALI tensor.
      """
      jax_array = jax.dlpack.from_dlpack(
          dali_tensor._expose_dlpack_capsule()
      )
  
      # For now we need this copy to make sure that underlying memory is available.
      # One solution is to implement full DLPack contract in DALI.
      # TODO(awolant): Remove this copy.
      return jax_array.copy()
  
label_desc: Use image node_id 1 for calling a global function with example usage: _to_jax_array(dali_tensor) and returns: jax_array