For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo module

reductions

`comptime` values

`logger`

comptime logger = Logger(stdout, prefix=String(""), source_location=False)

Structs

ApplyQKRMSNorm: Fused per-element QK-RMSNorm apply for two operands Q and K.
ArgMax:
ArgMin:
ArgNonZero:
ArgSort:
BottomK:
CumSum:
LayerNorm:
LogSoftmax:
Mean:
ReduceAdd:
ReduceGroupNorm:
ReduceMax:
ReduceMin:
ReduceMinAndMax:
ReduceMul:
ReduceRMSNorm:
ReduceRMSNormFusedResidualAdd:
ReduceRMSNormRoPE: Fuses RMS normalization and Rotary Position Embedding (RoPE) into one operation.
RowMeanOfSquares: Per-row mean of squares over the last axis, accumulated in float32.
RowMeanOfSquaresQK: Fused per-row mean of squares for two operands Q and K.
Softmax:
TopK:

Functions

arg_nonzero_shape:
bottom_k_shape:
composite_rms_norm_fused_residual_add_shape:
composite_rms_norm_rope_shape:
reduce_add_shape:
reduce_group_norm_shape:
reduce_layer_norm_shape:
reduce_max_shape:
reduce_mean_shape:
reduce_min_shape:
reduce_mul_shape:
reduce_reduce_min_and_max_shape:
reduce_rms_norm_shape:
reduce_row_mean_of_squares_qk_shape:
reduce_row_mean_of_squares_shape:
top_k_shape:

comptime values​

logger​

Structs​

Functions​

`comptime` values

`logger`

Structs

Functions