oa::ml::matrix::mamba3_mimo_step

Advance one Mamba-3 MIMO token and update four explicit recurrent states. State layouts are SSM [B,H,P,N], angle [B,H,A], key [B,H,R,N], and value [B,H,R,P]. Errors Returns an error unless config.sequence_length == 1, every operand and state matches the configured same-engine F32 layout, or recording fails.

FnRust public API

Function Documentation

pub fn mamba3_mimo_step(
	c: &Matrix,
	b: &Matrix,
	x: &Matrix,
	z: &Matrix,
	adt: &Matrix,
	dt: &Matrix,
	trap: &Matrix,
	angle: &Matrix,
	c_bias: &Matrix,
	b_bias: &Matrix,
	d: &Matrix,
	mimo_x: &Matrix,
	mimo_z: &Matrix,
	mimo_o: &Matrix,
	norm_weight: &Matrix,
	ssm_state: &mut Matrix,
	angle_state: &mut Matrix,
	k_state: &mut Matrix,
	v_state: &mut Matrix,
	config: SsmConfig
) -> Result<Matrix>