keras/src/optimizers/loss_scale_optimizer.py/LossScaleOptimizer/finalize_variable_values
class LossScaleOptimizer:    def finalize_variable_values(self, var_list):
        self.inner_optimizer.finalize_variable_values(var_list)