class_docstr: Wraps an OWL-ViT FLAX model for convenient inference.
  
  All public methods apply to a single example and take and return Numpy arrays.
  
  Attributes:
    config: ConfigDict with model configuration.
    module: OWL-ViT Flax module.
    variables: Variable dict to be used with module.apply.
code: def _predict_classes_jitted(
      self,
      image_features: jnp.ndarray,
      query_embeddings: jnp.ndarray,
  ) -> Dict[str, jnp.ndarray]:
      return self.module.apply(
          self.variables,
          image_features=image_features,
          query_embeddings=query_embeddings,
          method=self.module.class_predictor,
      )
  
imports: import dataclasses
  import functools
  from typing import Any, Dict, Tuple
  from flax import 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.notebooks import numpy_cache
  from scipy import special
  from skimage import transform
  import tensorflow
label_desc: Use image node_id 10 for calling the Model obj's underlying member method code with example usage: obj._predict_classes_jitted(image_features, query_embeddings) and returns: self