keras/src/callbacks/tensorboard.py/TensorBoard/_collect_learning_rate
class TensorBoard:    def _collect_learning_rate(self, logs):
        if isinstance(self.model.optimizer, Optimizer):
            logs["learning_rate"] = float(
                ops.convert_to_numpy(self.model.optimizer.learning_rate)
            )
        return logs