Module training.jax_loss
========================

.. contents::
    :local:

.. automodule:: training.jax_loss
.. currentmodule:: training.jax_loss








.. rubric:: Functions overview

.. autosummary::



  bounds_cost



  l0_norm_approx



  l2sqr_norm



  logsoftmax



  make_bounds



  mse



  softmax



  sse









.. rubric:: Functions
.. autofunction:: bounds_cost
.. autofunction:: l0_norm_approx
.. autofunction:: l2sqr_norm
.. autofunction:: logsoftmax
.. autofunction:: make_bounds
.. autofunction:: mse
.. autofunction:: softmax
.. autofunction:: sse