keras/src/layers/activations/softmax.py/Softmax/get_config
class Softmax:    def get_config(self):
        config = super().get_config()
        config.update({"axis": self.axis})
        return config