code: def next_power_of_2(x: int) -> int:
      if x == 0:
          return 1
      return int(2 ** math.ceil(math.log2(x)))
  
file_path: jax/jax/_src/pallas/utils.py
imports: import math
  import numpy
  from jax import lax
  from jax._src import core
  import jax.numpy
label_desc: Use image node_id 4 for calling a global function with example usage: next_power_of_2(x) and returns: int, int