optax.losses.softmax\_cross\_entropy
====================================

.. currentmodule:: optax.losses

.. autofunction:: softmax_cross_entropy