def METHOD_NAME(params: hk.Params, x, labels) -> jnp.ndarray:
    """Evaluation metrics (classification accuracy)."""
    logits = JaxHaikuServerAggregator.static_model.model_network.apply(params, x)
    predictions = jnp.argmax(logits, axis=-1)
    return jnp.mean(predictions == labels)