code: def _np2jnp(numpy_dict: Dict[str, np.ndarray]) -> Dict[str, Array]:
      for k, v in numpy_dict.items():
          numpy_dict[k] = jnp.array(v)
      return numpy_dict
  
file_path: safetensors/bindings/python/py_src/safetensors/flax.py
imports: import os
  from typing import Dict, Optional, Union
  import numpy
  import jax.numpy
  from jax import Array
  from safetensors import numpy, safe_open
label_desc: Use image node_id 5 for calling a global function with example usage: _np2jnp(numpy_dict) and returns: numpy_dict