code: def to_payload(
      cls,
      batch: jax.Array,
      batch_dim: int = 0,
  ) -> Payload:
      return cls.create_payload(
          pickle.dumps(np.asarray(batch)),
          batch.shape[batch_dim],
      )
  
file_path: BentoML/src/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 3 for calling the JaxArrayContainer obj's underlying member method code with example usage: obj.to_payload(cls, batch, batch_dim) and returns: cls