code: def helper_test_jitted_correctness(gen, train, train_jit):
      nojit = train(*gen()).numpy()
      for _ in range(5):
          jit = train_jit(*gen()).numpy()
      np.testing.assert_allclose(nojit, jit, rtol=1e-3, atol=1e-5)
  
file_path: tinygrad/test/external/external_test_jit_on_models.py
imports: import unittest
  import numpy
  from tinygrad import Tensor, dtypes
  from tinygrad.jit import TinyJit
  from tinygrad.helpers import CI
  from test.helpers import derandomize_model
  from examples.llama import Transformer
label_desc: Use image node_id 1 for calling a global function with example usage: helper_test_jitted_correctness(gen, train, train_jit) without return types