.. _losses: .. currentmodule:: mlx.nn.losses Loss Functions -------------- .. autosummary:: :toctree: _autosummary_functions :template: nn-module-template.rst binary_cross_entropy cosi
.. _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
Sign in to view the full prompt.
Sign In