keras/src/utils/tracking.py/TrackedDict/tree_unflatten
class TrackedDict:    def tree_unflatten(cls, keys, values):
        from keras.src.utils.module_utils import optree

        # For optree
        return cls(optree.utils.safe_zip(keys, values))