code: def _f(x):
      x = eqx.error_if(x, x < 0, "x must be non-negative")
      return jax.nn.relu(x)
  
file_path: equinox/tests/test_errors.py
imports: import equinox
  import equinox.internal
  import jax
  import jax.numpy
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: _f(x) and returns: jax