code: def clz(x):
      assert np.issubdtype(x.dtype, np.integer)
      nbits = np.iinfo(x.dtype).bits
      mask = (2 ** np.arange(nbits, dtype=x.dtype))[::-1]
      bits = (x[..., None] & mask).astype(np.bool_)
      out = np.argmax(bits, axis=-1).astype(x.dtype)
      out[x == 0] = nbits
      return out
  
imports: import builtins
  import collections
  import itertools
  import numpy
  import opt_einsum
  import scipy.special
  from jax._src import dtypes
  from jax._src import util
label_desc: Use image node_id 16 for calling a global function with example usage: clz(x) and returns: out