moving custom instructions

This commit is contained in:
Elias Bachaalany
2024-03-08 10:31:22 -08:00
parent a6ba81aa0e
commit f70541dcbe
1430 changed files with 852 additions and 1409 deletions

View File

@@ -0,0 +1,24 @@
.. _losses:
.. currentmodule:: mlx.nn.losses
Loss Functions
--------------
.. autosummary::
:toctree: _autosummary_functions
:template: nn-module-template.rst
binary_cross_entropy
cosine_similarity_loss
cross_entropy
gaussian_nll_loss
hinge_loss
huber_loss
kl_div_loss
l1_loss
log_cosh_loss
mse_loss
nll_loss
smooth_l1_loss
triplet_loss