mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
15 lines
162 B
ReStructuredText
15 lines
162 B
ReStructuredText
|
.. _fast:
|
||
|
|
|
||
|
|
Fast
|
||
|
|
====
|
||
|
|
|
||
|
|
.. currentmodule:: mlx.core.fast
|
||
|
|
|
||
|
|
.. autosummary::
|
||
|
|
:toctree: _autosummary
|
||
|
|
|
||
|
|
rms_norm
|
||
|
|
layer_norm
|
||
|
|
rope
|
||
|
|
scaled_dot_product_attention
|