What is the correct gradient at zero for `pointwise.unit_step`?
I just noticed that the gradient of the pointwise function unit_step is set to be always zero.
Is that correct, or should we return np.nan if the input is exactly zero, like for examples sign does?