mlex 0.1.2

Safe, idiomatic Rust runtime for Apple MLX: quantized LLM inference (Qwen, Gemma4, NemotronH, DharaAR, ...) with multi-modal support
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