docs update

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

@@ -0,0 +1,11 @@
+mlx.nn.hard\_shrink
+===================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: hard_shrink
+
+   
+
+   
+   

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

@@ -0,0 +1,11 @@
+mlx.nn.hard\_tanh
+=================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: hard_tanh
+
+   
+
+   
+   

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

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

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

@@ -17,6 +17,8 @@ simple functions.
    gelu_approx
    gelu_fast_approx
    glu
+   hard_shrink
+   hard_tanh
    hardswish
    leaky_relu
    log_sigmoid
@@ -29,6 +31,7 @@ simple functions.
    sigmoid
    silu
    softmax
+   softmin
    softplus
    softshrink
    step

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

@@ -21,10 +21,15 @@ Layers
    Dropout3d
    Embedding
    GELU
+   GLU
    GroupNorm
    GRU
+   HardShrink
+   HardTanh
+   Hardswish
    InstanceNorm
    LayerNorm
+   LeakyReLU
    Linear
    LSTM
    MaxPool1d
@@ -36,13 +41,19 @@ Layers
    QuantizedLinear
    RMSNorm
    ReLU
+   ReLU6
    RNN
    RoPE
    SELU
    Sequential
    SiLU
    SinusoidalPositionalEncoding
+   Softmin
    Softshrink
+   Softsign
+   Softmax
+   Softplus
    Step
+   Tanh
    Transformer
    Upsample