.. _layers:

.. currentmodule:: mlx.nn

Layers
------

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

   Sequential
   ReLU
   PReLU
   GELU
   SiLU
   Step
   SELU
   Mish
   Embedding
   Linear
   QuantizedLinear
   Conv1d
   Conv2d
   BatchNorm
   LayerNorm
   RMSNorm
   GroupNorm
   InstanceNorm
   Dropout
   Dropout2d
   Dropout3d
   Transformer
   MultiHeadAttention
   ALiBi
   RoPE
   SinusoidalPositionalEncoding
