Note to self: I should probably add a `np.clip` w/warning here...