code: def qr(x, /, *, mode="reduced"):
      """
      Returns the QR decomposition of a full column rank matrix (or a stack of matrices).
      """
      Q, R = jax.numpy.linalg.qr(x, mode=mode)
      return QRResult(Q=Q, R=R)
  
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 15 for calling a global function with example usage: qr() and returns: QRResult