code: def solve(x1, x2, /):
      """
      Returns the solution of a square system of linear equations with a unique solution.
      """
      if x2.ndim == 1:
          signature = "(m,m),(m)->(m)"
      else:
          signature = "(m,m),(m,n)->(m,n)"
      return jax.numpy.vectorize(
          jax.numpy.linalg.solve, signature=signature
      )(x1, x2)
  
imports: from functools import partial
  from typing import NamedTuple
  import jax
  from jax.experimental.array_api._data_type_functions import _promote_to_default_dtype
label_desc: Use image node_id 17 for calling a global function with example usage: solve() and returns: jax