code: def test_nested_jit_and_vmap(self, filter_mode):
      @jit
      def innermost(x):
          assert False
  
      @jit
      def inbetween(x):
          return 1 + vmap(innermost)(x)
  
      @jit
      def outermost(x):
          return 2 + inbetween(x)
  
      f = lambda: outermost(jnp.array([1, 2]))
  
      check_filtered_stack_trace(
          self,
          AssertionError,
          f,
          [
              ("<lambda>", "f = lambda: outermost"),
              ("outermost", "return 2 + inbetween(x)"),
              ("inbetween", "return 1 + vmap(innermost)(x)"),
              ("innermost", "assert False"),
          ],
          filter_mode=filter_mode,
      )
  
imports: import re
  import sys
  import traceback
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  import jax.numpy
  from jax import grad, jit, vmap, lax
  from jax._src import config
  from jax._src import core
  from jax._src import source_info_util
  from jax._src import test_util
  from jax._src import traceback_util
label_desc: Use image node_id 2 for calling the FilteredTracebackTest obj's underlying member method code with example usage: obj.test_nested_jit_and_vmap(filter_mode) and returns: unknown, unknown