https://github.com/okx/plonky2/blob/b74c2ac48bd25df0cbdee2e4b3871f63b631178d/plonky2/src/plonk/prover.rs#L306-L318