code: def relu(x: JaxArray) -> JaxArray:
      """Rectified linear unit activation function.
  
      Args:
          x: input tensor.
  
      Returns:
          tensor with the element-wise output relu(x) = max(x, 0).
      """
      return jax.nn.relu(x)
  
imports: import jax.nn
  import jax.scipy.special
  from jax import lax
  from objax.typing import JaxArray
label_desc: Use image node_id 1 for calling a global function with example usage: relu(x) and returns: jax