mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 09:51:17 +08:00
29 lines
317 B
ReStructuredText
29 lines
317 B
ReStructuredText
|
.. _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
|