Add training metrics that allows optimizers to summarize important metrics.
Add training metrics that allows optimizers to summarize important metrics.

PiperOrigin-RevId: 423580058

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo