IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /max/get-started.md). For the complete documentation index, see llms.txt.

Skip to main content

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

selective_scan_ops

Selective scan operation registrations for Mamba SSM.

This module registers the following ops:

selective_scan_fwd: Full selective scan forward pass
selective_scan_fwd_minimal: Minimal variant without optional tensors
selective_scan_update: Single-step update for autoregressive inference

Structs

SelectiveScanFwd: Selective scan forward pass operation for Mamba SSM.
SelectiveScanFwdMinimal: Minimal selective scan forward pass - no optional D, z, or delta_bias.
SelectiveScanUpdate: Selective scan update operation for autoregressive inference.

Structs