I was wondering if we had Power Iteration way of doing it !
But If I do use torch.svd(), will it be auto diffrentiable , I mean, If I add this as a regularizer (ex: in place of l2 loss), Since  calculating max singular value, through this way would not be diffrentiable. And we need it to be diffrentiable, since this would be used in calculating loss and updating.