keras/src/distribution/distribution_lib.py/LayoutMap/__delitem__
class LayoutMap:    def __delitem__(self, key):
        # let the dict to handle the key missing error
        return self._layout_map.pop(key)