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