mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-06 17:31:13 +08:00
18 lines
260 B
ReStructuredText
18 lines
260 B
ReStructuredText
.. _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
|