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

varlen_causal_conv1d_ops

Varlen Causal Conv1D operation registrations for Mamba SSM.

This module registers operations for variable-length causal 1D convolution:

causal_conv1d_varlen_fwd: Forward pass for varlen sequences
causal_conv1d_varlen_update: Update for decode/autoregressive inference
causal_conv1d_varlen_states: Extract states from varlen sequences

Structs

CausalConv1DVarlenFwd: Varlen causal 1D convolution forward pass.
CausalConv1DVarlenStates: Extract conv states from variable-length sequences.
CausalConv1DVarlenUpdate: Varlen causal conv1d update for autoregressive decoding.

Structs