MLAD beta | Prompts
← Prompts
Reference / Understand System Prompts Collection

.. currentmodule:: mlx.nn

.. _layers: .. currentmodule:: mlx.nn Layers ------ .. autosummary:: :toctree: _autosummary :template: nn-module-template.rst ALiBi BatchNorm Conv1d Conv2d Dropout Dropout2

.. _layers:

.. currentmodule:: mlx.nn

Layers
------

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

   ALiBi
   BatchNorm
   Conv1d
   Conv2d
   Dropout
   Dropout2d
   Dropout3d
   Embedding
   GELU
   GroupNorm
   InstanceNorm
   LayerNorm
   Linear
   Mish

Sign in to view the full prompt.

Sign In

Tags

openai

Classification

Reference Documentation, cheatsheets, setup guides
Reference Understand
Explain or analyze
Scope Project
This codebase
Manual Manually placed / Persistent