Function train_step_rmsprop 

pub fn train_step_rmsprop(
    graph: &mut Graph,
    optimizer: &mut RmsProp,
    prediction: NodeId,
    target: NodeId,
    trainable_nodes: &[NodeId],
) -> Result<f32, ModelError>

Runs one full train step: loss forward, backward, and RMSProp updates.