docs update

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

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

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

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

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

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

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

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

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

@@ -15,9 +15,9 @@ simple functions.
    gelu
    gelu_approx
    gelu_fast_approx
-   relu
+   mish
    prelu
+   relu
+   selu
    silu
    step
-   selu
-   mish

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

@@ -9,29 +9,29 @@ Layers
    :toctree: _autosummary
    :template: nn-module-template.rst
 
-   Sequential
-   ReLU
-   PReLU
-   GELU
-   SiLU
-   Step
-   SELU
-   Mish
-   Embedding
-   Linear
-   QuantizedLinear
+   ALiBi
+   BatchNorm
    Conv1d
    Conv2d
-   BatchNorm
-   LayerNorm
-   RMSNorm
-   GroupNorm
-   InstanceNorm
    Dropout
    Dropout2d
    Dropout3d
-   Transformer
+   Embedding
+   GELU
+   GroupNorm
+   InstanceNorm
+   LayerNorm
+   Linear
+   Mish
    MultiHeadAttention
-   ALiBi
+   PReLU
+   QuantizedLinear
+   RMSNorm
+   ReLU
    RoPE
+   SELU
+   Sequential
+   SiLU
    SinusoidalPositionalEncoding
+   Step
+   Transformer

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

@@ -10,14 +10,14 @@ Loss Functions
    :template: nn-module-template.rst
 
    binary_cross_entropy
+   cosine_similarity_loss
    cross_entropy
+   hinge_loss
+   huber_loss
    kl_div_loss
    l1_loss
+   log_cosh_loss
    mse_loss
    nll_loss
    smooth_l1_loss
-   triplet_loss
-   hinge_loss
-   huber_loss
-   log_cosh_loss
-   cosine_similarity_loss
+   triplet_loss

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

@@ -51,6 +51,10 @@ Operations
    greater_equal
    identity
    inner
+   isnan
+   isposinf
+   isneginf
+   isinf
    less
    less_equal
    linspace

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

@@ -33,13 +33,13 @@ we use a splittable version of Threefry, which is a counter-based PRNG.
 .. autosummary:: 
   :toctree: _autosummary
 
-   seed
-   key
-   split
    bernoulli
    categorical
    gumbel
+   key
    normal
    randint
-   uniform
+   seed
+   split
    truncated_normal
+   uniform