Make SimdMeshImpl gin configurable and set allreduce_in_bfloat16_max_group_size higher for evaluation.
Make SimdMeshImpl gin configurable and set allreduce_in_bfloat16_max_group_size higher for evaluation.

PiperOrigin-RevId: 329829913

google-research/text-to-text-transfer-transformer