code: def power(x: Array, p: float) -> Array:
      """Power transform; `power_tx(_, 1/p)` is the inverse of `power_tx(_, p)`."""
      chex.assert_type(x, float)
      q = jnp.sqrt(p)
      return jnp.sign(x) * (jnp.power(jnp.abs(x) / q + 1.0, p) - 1) / q
  
file_path: rlax/rlax/_src/transforms.py
imports: import chex
  import jax
  import jax.numpy
  from rlax._src import base
label_desc: Use image node_id 10 for calling a global function with example usage: power(x, p) and returns: unknown