Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
Warning
This page was created from a pull request (#9655).
Singular value decomposition.
Returns the singular values if compute_uv is False, otherwise returns a triple containing the left singular vectors, the singular values and the adjoint of the right singular vectors.
previous
jax.lax.linalg.qr
next
jax.lax.linalg.triangular_solve