code: def batch_to_payloads(
      cls,
      batch: jax.Array,
      indices: t.Sequence[int],
      batch_dim: int = 0,
  ) -> t.List[Payload]:
      batches = cls.batch_to_batches(batch, indices, batch_dim)
      payloads = [
          cls.to_payload(subbatch, batch_dim) for subbatch in batches
      ]
      return payloads
  
file_path: BentoML/grpc-client/bentoml/_internal/frameworks/common/jax.py
imports: from __future__ import annotations
  import itertools
  import pickle
  import typing
  from typing import TYPE_CHECKING
  from simple_di import inject
  from ....exceptions import MissingDependencyException
  from ...runner.container import DataContainer
  from ...runner.container import DataContainerRegistry
  from ...runner.container import Payload
  from ...types import LazyType
  from ...utils import LazyLoader
label_desc: Use image node_id 5 for calling the JaxArrayContainer obj's underlying member method code with example usage: obj.batch_to_payloads(cls, batch, indices, batch_dim) and returns: payloads