docs update

This commit is contained in:
Awni Hannun
2024-03-14 12:46:45 -07:00
committed by CircleCI Docs
parent fd78c54288
commit eede93197d
411 changed files with 15444 additions and 13066 deletions

View File

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

View File

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

View File

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

View File

@@ -21,9 +21,11 @@ Layers
Embedding
GELU
GroupNorm
GRU
InstanceNorm
LayerNorm
Linear
LSTM
MaxPool1d
MaxPool2d
Mish
@@ -32,6 +34,7 @@ Layers
QuantizedLinear
RMSNorm
ReLU
RNN
RoPE
SELU
Sequential