core.ssm.ops.common

Triton kernels shared by every linear-attention / SSM mixer.

Nothing here is specific to a recurrence. The short causal convolution that front-runs the recurrence, the slot-indexed state gather/scatter used by prefix caching, and the autotune-configuration filtering that makes the kernels deterministic are all common to Mamba2, Gated Delta Product and their relatives. Recurrence-specific kernels live in the sibling mamba2 and gdp packages.

Submodules

• core.ssm.ops.common.determinism
• core.ssm.ops.common.intermediate_extraction
• core.ssm.ops.common.causal_conv1d_varlen
• core.ssm.ops.common.causal_conv1d_triton