code: def matrix_transpose(x: ArrayLike, /) -> Array:
      """Transposes a matrix (or a stack of matrices) x."""
      check_arraylike("jnp.linalg.matrix_transpose", x)
      x_arr = jnp.asarray(x)
      ndim = x_arr.ndim
      if ndim < 2:
          raise ValueError(
              f"matrix_transpose requres at least 2 dimensions; got {ndim=}"
          )
      return jax.lax.transpose(
          x_arr, (*range(ndim - 2), ndim - 1, ndim - 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 38 for calling a global function with example usage: matrix_transpose() and returns: jax