def __call__(self, value: Any) -> Any:
    """Runs each of the sublayers in sequence.

    Args:
      value: The input to the layer.

    Returns:
      The output of the final sublayer.
    """
    for i, layer in enumerate(self.sublayers):
      with jax.named_scope(f"{i}"):
        value = layer(value)
    return value
Runs each of the sublayers in sequence.

    Args:
      value: The input to the layer.

    Returns:
      The output of the final sublayer.
    
penzai/deprecated/v1/nn/grouping.py