keras/src/layers/normalization/unit_normalization.py/UnitNormalization/get_config
class UnitNormalization:    def get_config(self):
        config = super().get_config()
        config.update({"axis": self.axis})
        return config