code: def make_mesh(
      topo: TopologyDescription,
      mesh_shape: Sequence[int],
      axis_names: tuple[str, ...],
      *,
      contiguous_submeshes: bool = False,
  ) -> jax.sharding.Mesh:
      devices = mesh_utils.create_device_mesh(
          mesh_shape,
          list(topo.devices),
          contiguous_submeshes=contiguous_submeshes,
      )
      return jax.sharding.Mesh(devices, axis_names)
  
file_path: jax/jax/experimental/topologies.py
imports: from __future__ import annotations
  import abc
  from collections.abc import Sequence
  from typing import Optional
  import jax
  from jax.experimental import mesh_utils
  from jax._src.lib import xla_client
  from jax._src import xla_bridge
label_desc: Use image node_id 3 for calling a global function with example usage: make_mesh(topo, mesh_shape, axis_names) and returns: jax