code: def get(tag_like: str | Tag) -> bentoml.Model:
      """
      Get the BentoML model with the given tag.
  
      Args:
          tag_like: The tag of the model to retrieve from the model store.
  
      Returns:
          :obj:`~bentoml.Model`: A BentoML :obj:`~bentoml.Model` with the matching tag.
  
      Example:
  
      .. code-block:: python
  
         import bentoml
  
         model = bentoml.flax.get("mnist:latest")
      """
      model = bentoml.models.get(tag_like)
      if model.info.module not in (MODULE_NAME, __name__):
          raise NotFound(
              f"Model {model.tag} was saved with module {model.info.module}, failed to load with {MODULE_NAME}."
          )
      return model
  
imports: from __future__ import annotations
  import functools
  import logging
  import typing
  from pickle import UnpicklingError
  from types import ModuleType
  from typing import TYPE_CHECKING
  import msgpack.exceptions
  import bentoml
  from ...exceptions import BentoMLException
  from ...exceptions import MissingDependencyException
  from ...exceptions import NotFound
  from ..models.model import ModelContext
  from ..models.model import PartialKwargsModelOptions
  from ..runner.utils import Params
  from ..types import LazyType
  from ..utils import LazyLoader
  from ..utils.pkg import get_pkg_version
  from .common.jax import JaxArrayContainer
  from .common.jax import jax
  from .common.jax import jnp
label_desc: Use image node_id 1 for calling a global function with example usage: get(tag_like) and returns: model