moving custom instructions

This commit is contained in:
Elias Bachaalany
2024-03-08 10:31:22 -08:00
parent a6ba81aa0e
commit f70541dcbe
1430 changed files with 852 additions and 1409 deletions

View File

@@ -0,0 +1,37 @@
.. _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
MultiHeadAttention
PReLU
QuantizedLinear
RMSNorm
ReLU
RoPE
SELU
Sequential
SiLU
SinusoidalPositionalEncoding
Step
Transformer