This is an old one, so sorry, if my question might be naive, but isn’t this valid for matrix calculation as well?

So, if I run this, I get the same result as with scipy:
X.inverse().sqrt()

I assume, this also maintains the gradients?