Function prove 

pub(crate) fn prove(
    params: &SystemParams,
    pk: &MultiStarkProvingKey<BabyBearPoseidon2Config>,
    cpu_traces: Vec<ColMajorMatrix<BabyBear>>,
) -> Proof<BabyBearPoseidon2Config>