.. _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