code: def logit(x: Array) -> Array:
      """Logit transform, inverse of sigmoid."""
      chex.assert_type(x, float)
      return -jnp.log(1.0 / x - 1.0)
  
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 3 for calling a global function with example usage: logit(x) and returns: unknown