code: def sigmoid(x: Numeric) -> Array:
      """Sigmoid transform."""
      chex.assert_type(x, float)
      return jax.nn.sigmoid(x)
  
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 2 for calling a global function with example usage: sigmoid(x) and returns: jax