The FAQ page says that the scipy.linalg submodule provides a more complete wrapper for the Fortran LAPACK library, while numpy.linalg tries to build independently of LAPACK.
I did some tests for various implementations of svd functions and found that scipy.linalg.svd is faster than a simple example:

However, jax is wrapped in NumPy, otherwise jax.numpy.linalg.svd even faster:

A full test pad is available here .
Zichen wang
source share