jax.scipy.linalg.inv

jax.scipy.linalg.inv(a, overwrite_a=False, check_finite=True)

Compute the inverse of a matrix.

LAX-backend implementation of scipy.linalg._basic.inv().

Does not support the Scipy argument check_finite=True, because compiled JAX code cannot perform checks of array values at runtime.

Does not support the Scipy argument overwrite_*=True.

Original docstring below.

Parameters:

• a (array_like) – Square matrix to be inverted.

• overwrite_a (bool)

• check_finite (bool)

Returns:

ainv – Inverse of the matrix a.

Return type:

ndarray