code: def test_args_kwargs():
      num_traces = 0
  
      @eqx.filter_jit
      def f(*args, **kwargs):
          nonlocal num_traces
          num_traces += 1
          return kwargs["x"]
  
      assert f(x=jnp.array(2)) == 2
      assert f(x=jnp.array(3)) == 3
      assert num_traces == 1
  
      assert f(x=jnp.array(3), y=jnp.array(4)) == 3
      assert num_traces == 2
  
      @eqx.filter_jit
      def h(*args, **kwargs):
          nonlocal num_traces
          num_traces += 1
          return args[0]
  
      assert h(1, 2) == 1
  
imports: import warnings
  from typing import Union
  import equinox
  import jax
  import jax.numpy
  import jax.random
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 4 for calling a global function with example usage: test_args_kwargs() and returns: kwargs, args