mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-21 08:41:13 +08:00
0fd2a1f4b0
* start * simple kernels working * restructure * inverse example working * docs + fixes * missing file * fix imports * address comments * add docs + fix test * Review comments + refactor to a single function * update docs * remove hashing * fix contig bug in test * back to a class * trailing whitespace * fix tests * match c++ and python apis * add link + make args kw_only
17 lines
195 B
ReStructuredText
17 lines
195 B
ReStructuredText
.. _fast:
|
|
|
|
Fast
|
|
====
|
|
|
|
.. currentmodule:: mlx.core.fast
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
rms_norm
|
|
layer_norm
|
|
rope
|
|
scaled_dot_product_attention
|
|
affine_quantize
|
|
metal_kernel
|