class_docstr: Abstract base class for all logit warpers that can be applied during generation with multinomial sampling.
docstr: Flax method for warping logits.
docstr_tok: ["Flax","method","for","warping","logits","."]
code: def __call__(
      self, input_ids: jnp.ndarray, scores: jnp.ndarray
  ) -> jnp.ndarray:
      """Flax method for warping logits."""
      raise NotImplementedError(
          f"{self.__class__} is an abstract class. Only classes inheriting this class can be called."
      )
  
imports: import inspect
  import jax
  import jax.lax
  import jax.numpy
  from ..utils import add_start_docstrings
  from ..utils.logging import get_logger
label_desc: Use image node_id 1 for calling the FlaxLogitsWarper obj's underlying member method code with example usage: obj.__call__(input_ids, scores) without return types