https://github.com/Lagrange-Labs/deep-prove/blob/master/zkml/src/model.rs#L446-L471 only handles all layers but convolution in a random fashion.