jax.lax.linalg.eig

jax.lax.linalg.eig(x, compute_left_eigenvectors=True, compute_right_eigenvectors=True)

Eigendecomposition of a general matrix.

Nonsymmetric eigendecomposition is at present only implemented on CPU.