code: def test_jit_kwargs(self):
      x = objax.random.normal((64, 3))
      kj = objax.Jit(LinearArgs(3, 3))
      y1 = kj(x, 1)
      y2 = kj(x, some_args=1)
      y3 = kj(x, some_args=2)
      self.assertEqual(y1.tolist(), y2.tolist())
      self.assertNotEqual(y1.tolist(), y3.tolist())
      kj = objax.Jit(LinearTrain(3, 3))
      with self.assertRaises(ConcretizationTypeError):
          kj(x, training=True)
  
file_path: objax/tests/jit.py
imports: import unittest
  import jax.numpy
  from jax.core import ConcretizationTypeError
  import objax
  from objax.typing import JaxArray
label_desc: Use image node_id 3 for calling the TestJit obj's underlying member method code with example usage: obj.test_jit_kwargs() without return types