code: def _assert_numpy_close(a, b, atol=None, rtol=None, err_msg=""):
      a, b = np.asarray(a), np.asarray(b)
      assert a.shape == b.shape
      atol = max(tolerance(a.dtype, atol), tolerance(b.dtype, atol))
      rtol = max(tolerance(a.dtype, rtol), tolerance(b.dtype, rtol))
      _assert_numpy_allclose(
          a, b, atol=atol * a.size, rtol=rtol * b.size, err_msg=err_msg
      )
  
imports: from functools import partial
  import operator
  from jax._src import api
  from jax._src import config
  from jax._src import dtypes
  from jax._src.tree_util import tree_map, tree_reduce
  import numpy
label_desc: Use image node_id 6 for calling a global function with example usage: _assert_numpy_close(a, b, atol, rtol, err_msg) without return types