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

conv

Registers convolution graph ops and dispatches them to the nn.conv kernels.

`comptime` values

`logger`

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

Structs

AvgPool: Registers the mo.avg_pool graph op with the graph compiler.
AvgPoolCeilModeTrue: Registers the mo.avg_pool_ceil_mode_true graph op with the graph compiler.
Conv: Registers the mo.conv graph op with the graph compiler.
Conv2dResidualAdd: Fused conv2d + TMA residual add + bias for SM100 (Blackwell).
ConvertE4M3FNToE4M3FNUZ: Registers the mo.convert_e4m3fn_to_e4m3fnuz graph op with the graph compiler.
ConvTranspose: Registers the mo.conv_transpose graph op with the graph compiler.
LayoutTransformFCQRS2FQRSCf: Registers the layout_transform_FCQRS_to_FQRSCf graph op with the graph compiler.
LayoutTransformFCRS2FRSCf: Registers the layout_transform_FCRS_to_FRSCf graph op with the graph compiler.
LayoutTransformQRSCF2FQRSCf: Registers the layout_transform_QRSCF_to_FQRSCf graph op with the graph compiler.
LayoutTransformQRSFC2FQRSCf: Registers the layout_transform_QRSFC_to_FQRSCf graph op with the graph compiler.
LayoutTransformRSCF2FRSCf: Registers the layout_transform_RSCF_to_FRSCf graph op with the graph compiler.
LayoutTransformRSFC2FRSCf: Registers the layout_transform_RSFC_to_FRSCf graph op with the graph compiler.
PackConvFilterShape: Registers the pack_conv_filter_shape graph op with the graph compiler.
PadConstant: Registers the mo.pad.constant graph op with the graph compiler.
PadReflect: Registers the mo.pad.reflect graph op with the graph compiler.
PadRepeat: Registers the mo.pad.repeat graph op with the graph compiler.

Functions

avg_pool_ceil_mode_true_shape: Computes the output shape for the mo.avg_pool_ceil_mode_true graph op.
avg_pool_shape: Computes the output shape for the mo.avg_pool graph op.
conv2d_residual_add_shape: Computes the output shape for the conv2d_residual_add graph op.
convert_e4m3fn_to_e4m3fnuz_shape: Computes the output shape for the mo.convert_e4m3fn_to_e4m3fnuz graph op.
mo_conv_shape: Computes the output shape for the mo.conv graph op.
mo_conv_transpose_shape: Computes the output shape for the mo.conv_transpose graph op.
pack_conv_filter_shape_fn: Compute the output shape of convolution filter packing.
pad_constant_shape: Computes the output shape for the mo.pad.constant graph op.
pad_reflect_shape: Computes the output shape for the mo.pad.reflect graph op.
pad_repeat_shape: Computes the output shape for the mo.pad.repeat graph op.

comptime values​

logger​

Structs​

Functions​

`comptime` values

`logger`

Structs

Functions