docs

docs/build/html/_sources/python/nn/_autosummary/mlx.nn.QuantizedLinear.rst

@@ -0,0 +1,8 @@
+mlx.nn.QuantizedLinear
+======================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: QuantizedLinear
+
+   

docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.rst

@@ -0,0 +1,8 @@
+mlx.nn.losses.smooth\_l1\_loss
+==============================
+
+.. currentmodule:: mlx.nn.losses
+
+.. autoclass:: smooth_l1_loss
+
+   

docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.rst

@@ -0,0 +1,8 @@
+mlx.nn.losses.triplet\_loss
+===========================
+
+.. currentmodule:: mlx.nn.losses
+
+.. autoclass:: triplet_loss
+
+   

docs/build/html/_sources/python/nn/layers.rst

@@ -26,3 +26,4 @@ Layers
    RoPE
    MultiHeadAttention
    Sequential
+   QuantizedLinear

docs/build/html/_sources/python/nn/losses.rst

@@ -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