{"id":"arxiv/2609.32505","url":"https://arxiv.org/abs/2609.32505","tags":["math.NA","cs.LG"],"score":0,"title":"Fast Differentiable SVD on GPU via Polar Decomposition","domain":"math.NA","source":"arxiv","authors":["Uliana Parkina","Askar Tsyganov","Sergei Kudriashov","Sergey Samsonov","Maxim Rakhuba"],"reviews":0,"subject":"math.NA","summary":"We present a fully GPU-oriented SVD pipeline based on polar decomposition, motivated by iterative methods that rely solely on matrix multiplications, such as the Newton-Schulz iteration. We show that this approach enables up to a $2\\times$ speedup compared to standard implementations. Furthermore, we derive a numerically stable backward pass for the polar decomposition and leverage it to obtain a fully differentiable SVD. Our methods are released as open-source implementations in both PyTorch and JAX: https://github.com/fallnlove/cans_svd.","category":"research","comments":0,"disputed":false,"division":"math","mentions":0,"discussed":0,"published":"2026-09-26","reactions":{}}