SVD grad values are 'NaN'

Hi all,
when dealing with matrices with number of rows much greater than number of columns (X10)
I’m receiving NaN grads for SVD

when I use matrix size of 256X15 everything is fine
when I use 4096X100 I get NaN (this is the size I need)
any idea why?

I have a similar problem. I also get NaN in the gradient. My matrix has the size of [2028, 64]. Please post the solutions if you fixed it.