optax.losses.kl\_divergence
===========================

.. currentmodule:: optax.losses

.. autofunction:: kl_divergence