Add justification for numerals in `sz`

[Demfier]

Hello @eladhoffer ,

As mentioned in the paper, sz needs to be a multiple of 4, which doesn't necessarily strike at first when we look at the equation. So I thought, adding the comment could be a bit useful.

Thank you for the implementation btw.

• Gaurav.