1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
.. _fast: Fast ==== .. currentmodule:: mlx.core.fast .. autosummary:: :toctree: _autosummary rms_norm layer_norm rope scaled_dot_product_attention metal_kernel cuda_kernel