code: def _is_at_level(obj, level):
      """Checks if `obj` is an at level `level`."""
      is_leaf = not isinstance(obj, (list, tuple, dict))
      if level == 0 or is_leaf:
          return (level == 0) == is_leaf
  
      if isinstance(obj, dict):
          elems = obj.values()
      else:
          elems = obj
      return elems and all(_is_at_level(x, level - 1) for x in elems)
  
file_path: trax/trax/fastmath/numpy.py
imports: import numpy
  from scipy.special import logsumexp
  from trax.shapes import signature
label_desc: Use image node_id 15 for calling a global function with example usage: _is_at_level(obj, level) and returns: unknown, unknown