flax

e5dc163d - Benchmarks class should scrap only the TF summary files and only scalar values from them.

Benchmarks class should scrap only the TF summary files and only scalar values from them. * IsSummaryEventsFile will filter out only the TF summary event files. So the benchmark output dir can contain other files e.g: checkpoints. * Scalar values have plugin name as "scalars". Hardcoding this is probably not the best of the idea (as TB may change it in the future?). https://github.com/tensorflow/tensorboard/blob/c053836702db370565465644f6d3b74e9e821711/tensorboard/plugins/scalar/metadata.py#L27 - the value is used from here and is not importable. Discussed with the TB team for the approach used here. PiperOrigin-RevId: 326610862