Module ssd 

The standard SSD kernels specialised to the single-pass scale/mask. Standard MIMO-first SSD kernels specialised to the single-pass form.

Three exact reformulations (Minimal / Serial / SerialRecalculated) that agree on values and gradients; selected via Mamba3SsdPath and dispatched by [Mamba3SingleSsdInput::run]. These take the raw v plus gamma/scale so the kernel applies the trapezoid weights and boundary seed internally.

Re-exports

pub use serial_recalculated::Mamba3SingleSsdAutodiffBackendExt;

pub use serial_recalculated::Mamba3SingleSsdBackendExt;

pub use ssd_path::Mamba3SingleSsdInput;

Modules

minimal

Matmul/segsum MIMO-first SSD with plain autodiff backward.

serial

Serial-over-chunks SSD with plain autodiff backward.

serial_recalculated

Serial-over-chunks SSD with a custom recompute backward. Single-SSD serial scan with a custom, memory-efficient backward.

ssd_path

The Mamba3SingleSsdInput bundle and its run() dispatch.