docs update

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

@@ -0,0 +1,6 @@
+mlx.nn.Upsample
+===============
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: Upsample

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

@@ -0,0 +1,6 @@
+mlx.nn.elu
+==========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: elu

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

@@ -0,0 +1,6 @@
+mlx.nn.glu
+==========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: glu

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

@@ -0,0 +1,6 @@
+mlx.nn.hardswish
+================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: hardswish

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

@@ -0,0 +1,6 @@
+mlx.nn.leaky\_relu
+==================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: leaky_relu

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

@@ -0,0 +1,6 @@
+mlx.nn.log\_sigmoid
+===================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: log_sigmoid

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

@@ -0,0 +1,6 @@
+mlx.nn.log\_softmax
+===================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: log_softmax

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

@@ -0,0 +1,6 @@
+mlx.nn.relu6
+============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: relu6

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

@@ -0,0 +1,6 @@
+mlx.nn.sigmoid
+==============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: sigmoid

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

@@ -0,0 +1,6 @@
+mlx.nn.softmax
+==============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: softmax

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

@@ -0,0 +1,6 @@
+mlx.nn.softplus
+===============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: softplus

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

@@ -0,0 +1,6 @@
+mlx.nn.tanh
+===========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: tanh

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

@@ -12,13 +12,24 @@ simple functions.
    :toctree: _autosummary_functions
    :template: nn-module-template.rst
 
+   elu
    gelu
    gelu_approx
    gelu_fast_approx
+   glu
+   hardswish
+   leaky_relu
+   log_sigmoid
+   log_softmax
    mish
    prelu
    relu
+   relu6
    selu
-   softshrink
+   sigmoid
    silu
+   softmax
+   softplus
+   softshrink
    step
+   tanh

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

@@ -40,3 +40,4 @@ Layers
    Softshrink
    Step
    Transformer
+   Upsample