Great, thank you! Very neat approach. Just one thing, running your example in parallel gives an error in np.delete(all_blocks, boundary_blocks), since the indices in boundary_blocks exceed the size of all_blocks.
@CastriMik this looks identical to what I did in the setDiagonal() variant, doesn’t it?