def unchecked_layer_call(
    func: Callable[[Layer, Any], Any],
) -> Callable[[Layer, Any], Any]:
  """Shape-checking opt-out decorator for ``Layer.__call__``."""
  func.__penzai_checked_layer_call__ = False
  return func
Shape-checking opt-out decorator for ``Layer.__call__``.
penzai/deprecated/v1/core/layer.py