docs update

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

@@ -55,6 +55,7 @@
       ~array.tolist
       ~array.transpose
       ~array.var
+      ~array.view
    
    
 

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

@@ -1,6 +0,0 @@
-mlx.core.block\_sparse\_mm
-==========================
-
-.. currentmodule:: mlx.core
-
-.. autofunction:: block_sparse_mm

docs/build/html/_sources/python/_autosummary/mlx.core.distributed.Group.rst

@@ -0,0 +1,25 @@
+mlx.core.distributed.Group
+==========================
+
+.. currentmodule:: mlx.core.distributed
+
+.. autoclass:: Group
+
+   
+   .. automethod:: __init__
+
+   
+   .. rubric:: Methods
+
+   .. autosummary::
+   
+      ~Group.__init__
+      ~Group.rank
+      ~Group.size
+      ~Group.split
+   
+   
+
+   
+   
+   

docs/build/html/_sources/python/_autosummary/mlx.core.distributed.all_gather.rst

@@ -0,0 +1,6 @@
+mlx.core.distributed.all\_gather
+================================
+
+.. currentmodule:: mlx.core.distributed
+
+.. autofunction:: all_gather

docs/build/html/_sources/python/_autosummary/mlx.core.distributed.all_sum.rst

@@ -0,0 +1,6 @@
+mlx.core.distributed.all\_sum
+=============================
+
+.. currentmodule:: mlx.core.distributed
+
+.. autofunction:: all_sum

docs/build/html/_sources/python/_autosummary/mlx.core.distributed.init.rst

@@ -0,0 +1,6 @@
+mlx.core.distributed.init
+=========================
+
+.. currentmodule:: mlx.core.distributed
+
+.. autofunction:: init

docs/build/html/_sources/python/_autosummary/mlx.core.distributed.is_available.rst

@@ -0,0 +1,6 @@
+mlx.core.distributed.is\_available
+==================================
+
+.. currentmodule:: mlx.core.distributed
+
+.. autofunction:: is_available

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

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

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

@@ -0,0 +1,19 @@
+.. _distributed:
+
+.. currentmodule:: mlx.core.distributed
+
+Distributed Communication
+==========================
+
+MLX provides a distributed communication package using MPI. The MPI library is
+loaded at runtime; if MPI is available then distributed communication is also
+made available.
+
+.. autosummary::
+   :toctree: _autosummary
+
+    Group
+    is_available
+    init
+    all_sum
+    all_gather

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

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

@@ -156,6 +156,7 @@ Operations
    tril
    triu
    var
+   view
    where
    zeros
    zeros_like