keras/src/metrics/probabilistic_metrics.py/KLDivergence/__init__
class KLDivergence:    def __init__(self, name="kl_divergence", dtype=None):
        super().__init__(fn=kl_divergence, name=name, dtype=dtype)