.. _layers:

.. currentmodule:: mlx.nn

Layers
------

.. autosummary::
   :toctree: _autosummary
   :template: nn-module-template.rst

   ALiBi
   AvgPool1d
   AvgPool2d
   BatchNorm
   Conv1d
   Conv2d
   Conv3d
   Dropout
   Dropout2d
   Dropout3d
   Embedding
   GELU
   GroupNorm
   GRU
   InstanceNorm
   LayerNorm
   Linear
   LSTM
   MaxPool1d
   MaxPool2d
   Mish
   MultiHeadAttention
   PReLU
   QuantizedEmbedding
   QuantizedLinear
   RMSNorm
   ReLU
   RNN
   RoPE
   SELU
   Sequential
   SiLU
   SinusoidalPositionalEncoding
   Softshrink
   Step
   Transformer
   Upsample