code: def roi_align_batch_serial(
      feature_map: jnp.ndarray, boxes: jnp.ndarray, output_width: int
  ) -> jnp.ndarray:
      """Applies RoIAlign serially to a batch of boxes."""
      roi_align_single = functools.partial(
          roi_align, feature_map, output_width=output_width
      )
      return jax.lax.map(roi_align_single, boxes)
  
imports: import abc
  import functools
  from typing import Any, Callable, Dict, Optional, Tuple, Union, Sequence
  from absl import logging
  from big_vision.models import bit
  from big_vision.models import vit
  import flax
  import flax.linen
  import jax
  import jax.numpy
  import ml_collections
  import numpy
  from scenic.model_lib.base_models import box_utils
  from scenic.projects.owl_vit import utils
  from scenic.projects.owl_vit.clip import layers
  from scenic.projects.owl_vit.clip import model
label_desc: Use image node_id 1 for calling a global function with example usage: roi_align_batch_serial(feature_map, boxes, output_width) and returns: jax