Adds pytypes.Metrics (Dict[str, clu_metrics.Metric]) to process_decode_out optionally, to allow metrics that need to run on the host from the start and/or need the input to be strings.
Adds pytypes.Metrics (Dict[str, clu_metrics.Metric]) to process_decode_out optionally, to allow metrics that need to run on the host from the start and/or need the input to be strings.

PiperOrigin-RevId: 463835119

google/praxis,google/praxis