.. _layers:

.. currentmodule:: mlx.nn

Layers
------

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

   Embedding
   ReLU
   PReLU
   GELU
   SiLU
   Step
   SELU
   Mish
   Linear
   Conv1d
   Conv2d
   LayerNorm
   RMSNorm
   GroupNorm
   RoPE
   MultiHeadAttention
   Sequential
   QuantizedLinear
   Dropout
   Dropout2d