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).
Eigendecomposition of a general matrix.
Nonsymmetric eigendecomposition is at present only implemented on CPU.
previous
jax.lax.linalg.cholesky
next
jax.lax.linalg.eigh