optimizer_loss_scale function

An optimizer that dynamically scales the loss to prevent underflow.