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.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!