code: def inv(a: ArrayLike) -> Array:
      check_arraylike("jnp.linalg.inv", a)
      arr = jnp.asarray(a)
      if arr.ndim < 2 or arr.shape[-1] != arr.shape[-2]:
          raise ValueError(
              f"Argument to inv must have shape [..., n, n], got {arr.shape}."
          )
      return solve(
          arr,
          lax.broadcast(
              jnp.eye(arr.shape[-1], dtype=arr.dtype), arr.shape[:-2]
          ),
      )
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  from functools import partial
  import numpy
  import textwrap
  import operator
  from typing import Literal, cast, overload
  import jax
  from jax import jit, custom_jvp
  from jax import lax
  from jax._src.lax import lax
  from jax._src.lax import linalg
  from jax._src.numpy import lax_numpy
  from jax._src.numpy import reductions, ufuncs
  from jax._src.numpy.util import _wraps, promote_dtypes_inexact, check_arraylike
  from jax._src.util import canonicalize_axis
  from jax._src.typing import ArrayLike, Array
label_desc: Use image node_id 27 for calling a global function with example usage: inv(a) and returns: solve