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
varlen_selective_scan_ops
Varlen selective scan operation registrations for Mamba SSM.
This module registers operations for variable-length selective scan:
- varlen_selective_scan_fwd: Forward pass for varlen sequences
- varlen_selective_state_update: State update for decode/autoregressive inference
comptime values
Strides1D
comptime Strides1D = IndexList[1]
Strides2D
comptime Strides2D = IndexList[2]
Strides3D
comptime Strides3D = IndexList[3]
Strides4D
comptime Strides4D = IndexList[4]
Structs
-
VarlenSelectiveScanFwd: Variable-length selective scan forward pass. -
VarlenSelectiveStateUpdate: Varlen selective state update for autoregressive inference.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!