This commit is contained in:
Awni Hannun
2023-12-17 13:23:03 -08:00
committed by CircleCI Docs
parent f75712551d
commit 0250e203f6
319 changed files with 39952 additions and 21579 deletions

View File

@@ -0,0 +1,17 @@
.. _losses:
.. currentmodule:: mlx.nn.losses
Loss Functions
--------------
.. autosummary::
:toctree: _autosummary_functions
:template: nn-module-template.rst
cross_entropy
binary_cross_entropy
l1_loss
mse_loss
nll_loss
kl_div_loss