keras/src/utils/tracking.py/TrackedDict/update
class TrackedDict:    def update(self, mapping):
        if self.tracker:
            mapping = {k: self.tracker.track(v) for k, v in mapping.items()}
        super().update(mapping)