code: def eye(n_rows, n_cols=None, /, *, k=0, dtype=None, device=None):
      return jax.device_put(
          jnp.eye(n_rows, n_cols, k=k, dtype=dtype), device=device
      )
  
file_path: jax/jax/experimental/array_api/_creation_functions.py
imports: import jax
  import jax.numpy
label_desc: Use image node_id 5 for calling a global function with example usage: eye() and returns: jax