code: def shard_alike_lowering(ctx, x, y):
      return _group_shard(ctx, x, y, *ctx.avals_out)
  
file_path: jax/jax/_src/shard_alike.py
imports: from functools import partial
  import itertools
  from jax._src import core
  from jax._src.interpreters import ad
  from jax._src.interpreters import mlir
  from jax._src.dispatch import apply_primitive
  from jax._src.tree_util import tree_flatten, tree_unflatten
  from jax._src.interpreters import batching
  from jax._src.util import safe_zip
  from jax._src.lib import xla_client
  from jax._src.api_util import shaped_abstractify
  from jax._src.lib.mlir import ir
label_desc: Use image node_id 5 for calling a global function with example usage: shard_alike_lowering(ctx, x, y) and returns: _group_shard