docs update

docs/build/html/_sources/python/_autosummary/mlx.core.linalg.cross.rst

@@ -0,0 +1,6 @@
+mlx.core.linalg.cross
+=====================
+
+.. currentmodule:: mlx.core.linalg
+
+.. autofunction:: cross

docs/build/html/_sources/python/_autosummary/mlx.core.put_along_axis.rst

@@ -0,0 +1,6 @@
+mlx.core.put\_along\_axis
+=========================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: put_along_axis

docs/build/html/_sources/python/linalg.rst

@@ -13,5 +13,6 @@ Linear Algebra
     norm
     cholesky
     cholesky_inv
+    cross
     qr
     svd

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

@@ -0,0 +1,16 @@
+mlx.nn.CELU
+===========
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: CELU
+
+   
+
+   
+   .. rubric:: Methods
+
+   .. autosummary::
+   
+   
+   

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

@@ -0,0 +1,16 @@
+mlx.nn.ELU
+==========
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: ELU
+
+   
+
+   
+   .. rubric:: Methods
+
+   .. autosummary::
+   
+   
+   

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

@@ -0,0 +1,16 @@
+mlx.nn.LogSigmoid
+=================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: LogSigmoid
+
+   
+
+   
+   .. rubric:: Methods
+
+   .. autosummary::
+   
+   
+   

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

@@ -0,0 +1,16 @@
+mlx.nn.LogSoftmax
+=================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: LogSoftmax
+
+   
+
+   
+   .. rubric:: Methods
+
+   .. autosummary::
+   
+   
+   

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

@@ -0,0 +1,16 @@
+mlx.nn.Sigmoid
+==============
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: Sigmoid
+
+   
+
+   
+   .. rubric:: Methods
+
+   .. autosummary::
+   
+   
+   

docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.celu.rst

@@ -0,0 +1,11 @@
+mlx.nn.celu
+===========
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: celu
+
+   
+
+   
+   

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

@@ -13,6 +13,7 @@ simple functions.
    :template: nn-module-template.rst
 
    elu
+   celu
    gelu
    gelu_approx
    gelu_fast_approx

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

@@ -13,6 +13,7 @@ Layers
    AvgPool1d
    AvgPool2d
    BatchNorm
+   CELU
    Conv1d
    Conv2d
    Conv3d
@@ -23,6 +24,7 @@ Layers
    Dropout2d
    Dropout3d
    Embedding
+   ELU
    GELU
    GLU
    GroupNorm
@@ -34,6 +36,8 @@ Layers
    LayerNorm
    LeakyReLU
    Linear
+   LogSigmoid
+   LogSoftmax
    LSTM
    MaxPool1d
    MaxPool2d
@@ -49,6 +53,7 @@ Layers
    RoPE
    SELU
    Sequential
+   Sigmoid
    SiLU
    SinusoidalPositionalEncoding
    Softmin

docs/build/html/_sources/python/ops.rst

@@ -121,6 +121,7 @@ Operations
    pad
    power
    prod
+   put_along_axis
    quantize
    quantized_matmul
    radians