optax.losses.convex\_kl\_divergence
===================================

.. currentmodule:: optax.losses

.. autofunction:: convex_kl_divergence