class_docstr: JIT (Just-In-Time) module takes a function or a module and compiles it for faster execution.
class_docstr_tok: ["JIT","(","Just-In-Time",")","module","takes","a","function","or","a","module","and","compiles","it","for","faster","execution","."]
docstr: Call the compiled version of the function or module.
docstr_tok: ["Call","the","compiled","version","of","the","function","or","module","."]
code: def __call__(self, *args, **kwargs):
      """Call the compiled version of the function or module."""
      output, changes = self._call(self.vc.tensors(), kwargs, *args)
      self.vc.assign(changes)
      return output
  
imports: from collections import namedtuple
  from typing import Optional, List, Union, Callable, Tuple
  import jax
  import jax.numpy
  import numpy
  from objax.typing import JaxArray
  from objax.util import class_name, override_args_kwargs, positional_args_names, repr_function
  from objax.variable import BaseVar, RandomState, VarCollection
label_desc: Use image node_id 2 for calling the Jit obj's underlying member method code with example usage: obj.__call__() and returns: output