code: def batch_np(
      graphs: Sequence[gn_graph.GraphsTuple],
  ) -> gn_graph.GraphsTuple:
      """Numpy implementation of `batch`. See `batch` for more details."""
      return _batch(graphs, np_=np)
  
imports: import functools
  from typing import Any, Callable, Generator, Iterable, Iterator, List, Mapping, Optional, Sequence, Union
  import jax
  from jax import lax
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  import numpy
label_desc: Use image node_id 13 for calling a global function with example usage: batch_np(graphs) and returns: _batch