pub trait LayerState: Send + Sync {
// Required methods
fn as_any(&self) -> &dyn Any;
fn as_any_mut(&mut self) -> &mut dyn Any;
}Expand description
Per-layer persistent state tracked across decode steps.
Attention layers use EmptyLayerState (KV lives in PagedKvCache).
SSM layers use SsmLayerState (recurrent h_state + conv_state).
Custom layers can implement this trait for arbitrary state.