docs update

This commit is contained in:
Awni Hannun
2024-09-28 11:04:59 -07:00
committed by CircleCI Docs
parent 6a6ffb598d
commit a68317ae17
2077 changed files with 14038 additions and 303005 deletions

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

@@ -13,6 +13,7 @@ simple functions.
:template: nn-module-template.rst
elu
celu
gelu
gelu_approx
gelu_fast_approx

View File

@@ -13,6 +13,7 @@ Layers
AvgPool1d
AvgPool2d
BatchNorm
CELU
Conv1d
Conv2d
Conv3d
@@ -23,6 +24,7 @@ Layers
Dropout2d
Dropout3d
Embedding
ELU
GELU
GLU
GroupNorm
@@ -34,6 +36,8 @@ Layers
LayerNorm
LeakyReLU
Linear
LogSigmoid
LogSoftmax
LSTM
MaxPool1d
MaxPool2d
@@ -49,6 +53,7 @@ Layers
RoPE
SELU
Sequential
Sigmoid
SiLU
SinusoidalPositionalEncoding
Softmin