Hey folks, we need to custom the adder and multiplier as well. We are currently trying the Sami’s code snippet (loopy version) and we are suffering due to the extremely long run time. As Sami mentioned above, it would be faster to use numpy instead of the loopy version. Do we still need to a forloop to unloop kernels_flat and imageunfold? Some advice would help us a lot. Thanks!