docs/src/python/nn/losses.rst

.. _losses:

.. currentmodule:: mlx.nn.losses

Loss Functions
--------------

.. autosummary::
   :toctree: _autosummary_functions
   :template: nn-module-template.rst

   binary_cross_entropy
   cross_entropy
   kl_div_loss
   l1_loss
   mse_loss
   nll_loss
   smooth_l1_loss
   triplet_loss
   hinge_loss
   huber_loss
   log_cosh_loss
   cosine_similarity_loss
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-17 13:20:55 -08:00			`.. _losses:`

			`.. currentmodule:: mlx.nn.losses`

			`Loss Functions`
			`--------------`

			`.. autosummary::`
			`:toctree: _autosummary_functions`
			`:template: nn-module-template.rst`

			`binary_cross_entropy`
Add smoothed L1 loss and enhancements to cross entropy loss (#166) * Add smooth_l1_loss * Add labels moothing for cross entropy loss --------- Co-authored-by: Awni Hannun <awni@apple.com> 2023-12-18 23:26:21 +08:00			`cross_entropy`
			`kl_div_loss`
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-17 13:20:55 -08:00			`l1_loss`
			`mse_loss`
			`nll_loss`
Add smoothed L1 loss and enhancements to cross entropy loss (#166) * Add smooth_l1_loss * Add labels moothing for cross entropy loss --------- Co-authored-by: Awni Hannun <awni@apple.com> 2023-12-18 23:26:21 +08:00			`smooth_l1_loss`
Add Hinge, Huber and LogCosh losses (#199) 2023-12-22 19:28:10 +01:00			`triplet_loss`
			`hinge_loss`
			`huber_loss`
Additoinal losses (#336) * cosine similarity loss --------- Co-authored-by: Awni Hannun <awni@apple.com> * Docstring nits 2024-01-09 07:01:13 +09:00			`log_cosh_loss`
			`cosine_similarity_loss`