From 606b3850c702a37170b4f8ef3d230bb7de1ff43e Mon Sep 17 00:00:00 2001
From: winklemad <winklemad@outlook.com>
Date: Wed, 15 Jul 2026 14:05:14 +0530
Subject: [PATCH] Fix log_cosh_loss docstring to document the element-wise loss

---
 python/mlx/nn/losses.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/python/mlx/nn/losses.py b/python/mlx/nn/losses.py
index 61b6adb047..f402f0298f 100644
--- a/python/mlx/nn/losses.py
+++ b/python/mlx/nn/losses.py
@@ -504,8 +504,7 @@ def log_cosh_loss(
     .. math::
 
        \text{logcosh}(y_{\text{true}}, y_{\text{pred}}) =
-            \frac{1}{n} \sum_{i=1}^{n}
-            \log(\cosh(y_{\text{pred}}^{(i)} - y_{\text{true}}^{(i)}))
+            \log(\cosh(y_{\text{pred}} - y_{\text{true}}))
 
 
     Args: