Skip to main content

Module infer

Module infer 

Source
Expand description

FlowState-R1 inference on the Burn backend (Flex CPU) for tensor stages, with the original verbatim host kernels for the SSM core.

Projections (embed, B/C, decoder, basis), activations and norms run on Burn tensors; the diagonal SSM scan (ssm_scan_step), discretization math, RevIN, Legendre basis and quantile recalibration are engine-agnostic host code. Weight loading reads GGUF straight via zsfm_burn::GgufWeights.

Structsยง

BurnFlowStateModel
InferConfig