docs

2025-10-17 19:14:54 +00:00
parent 20a3e22ff0
commit 6f71a74c87
273 changed files with 13936 additions and 1646 deletions
@@ -0,0 +1,8 @@
+mlx.nn.QuantizedLinear
+======================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: QuantizedLinear
+
+   
@@ -0,0 +1,8 @@
+mlx.nn.losses.smooth\_l1\_loss
+==============================
+
+.. currentmodule:: mlx.nn.losses
+
+.. autoclass:: smooth_l1_loss
+
+   
@@ -0,0 +1,8 @@
+mlx.nn.losses.triplet\_loss
+===========================
+
+.. currentmodule:: mlx.nn.losses
+
+.. autoclass:: triplet_loss
+
+   
@@ -26,3 +26,4 @@ Layers
   RoPE
   MultiHeadAttention
   Sequential
+   QuantizedLinear
@@ -9,9 +9,11 @@ Loss Functions
   :toctree: _autosummary_functions
   :template: nn-module-template.rst

-   cross_entropy
   binary_cross_entropy
+   cross_entropy
+   kl_div_loss
   l1_loss
   mse_loss
   nll_loss
-   kl_div_loss
+   smooth_l1_loss
+   triplet_loss