Scale reporting the loss when aggregate_losses_across_replicas=True.
Scale reporting the loss when aggregate_losses_across_replicas=True.

PiperOrigin-RevId: 483823503
Change-Id: If4a7c5d296208b162b3d12e9b23f6803fca865f3

google-research/circuit_training