code: def test_grad():
      g = jax.grad(_f)
      jg = jax.jit(g)
  
      for h in (g, jg):
          h(1.0)
          with _error:
              h(-1.0)
  
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 5 for calling a global function with example usage: test_grad() without return types