code: def tree_allclose(x, y, *, rtol=1e-5, atol=1e-8):
      return eqx.tree_equal(x, y, typematch=True, rtol=rtol, atol=atol)
  
imports: import equinox
  import jax.random
  import jax.tree_util
label_desc: Use image node_id 2 for calling a global function with example usage: tree_allclose(x, y) and returns: eqx