def METHOD_NAME(x):
    return jnp.where(
        x < -37.0,
        jnp.exp(x),
        jnp.where(
            x < 18.0,
            jnp.log1p(jnp.exp(x)),
            jnp.where(
                x < 33.3,
                x + jnp.exp(-x),
                x,
            ),
        ),
    )