docs update

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

@@ -0,0 +1,6 @@
+mlx.core.atleast\_1d
+====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: atleast_1d

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

@@ -0,0 +1,6 @@
+mlx.core.atleast\_2d
+====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: atleast_2d

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

@@ -0,0 +1,6 @@
+mlx.core.atleast\_3d
+====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: atleast_3d

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

@@ -0,0 +1,6 @@
+mlx.core.conv\_general
+======================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: conv_general

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

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

@@ -25,6 +25,9 @@ Operations
    argpartition
    argsort
    array_equal
+   atleast_1d
+   atleast_2d
+   atleast_3d
    broadcast_to
    ceil
    clip
@@ -32,6 +35,7 @@ Operations
    convolve
    conv1d
    conv2d
+   conv_general
    cos
    cosh
    dequantize

docs/build/html/_sources/python/optimizers/_autosummary/mlx.optimizers.join_schedules.rst

@@ -0,0 +1,6 @@
+mlx.optimizers.join\_schedules
+==============================
+
+.. currentmodule:: mlx.optimizers
+
+.. autofunction:: join_schedules

docs/build/html/_sources/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.rst

@@ -0,0 +1,6 @@
+mlx.optimizers.linear\_schedule
+===============================
+
+.. currentmodule:: mlx.optimizers
+
+.. autofunction:: linear_schedule

docs/build/html/_sources/python/optimizers/schedulers.rst

@@ -8,6 +8,8 @@ Schedulers
 .. autosummary::
    :toctree: _autosummary
 
-   step_decay    
-   exponential_decay    
    cosine_decay    
+   exponential_decay    
+   join_schedules
+   linear_schedule
+   step_decay