mlx/docs/src/python/fast.rst

.. _fast:

Fast
====

.. currentmodule:: mlx.core.fast

.. autosummary:: 
  :toctree: _autosummary

  rms_norm
  layer_norm
  rope
  scaled_dot_product_attention
  metal_kernel
  cuda_kernel
post nanobind docs fixes and some updates (#889) * post nanobind docs fixes and some updates * one more doc nit * fix for stubs and latex 2024-03-24 15:03:27 -07:00			`.. _fast:`

			`Fast`
			`====`

			`.. currentmodule:: mlx.core.fast`

			`.. autosummary::`
			`:toctree: _autosummary`

			`rms_norm`
			`layer_norm`
			`rope`
			`scaled_dot_product_attention`
Custom Metal Kernels from Python (#1325) * 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 2024-08-22 13:46:29 -07:00			`metal_kernel`
Custom cuda kernel (#2517) 2025-08-20 17:20:22 -07:00			`cuda_kernel`