code: def test_nested_jit_stack(self):
      @jax.grad
      @jax.jit
      def f(x):
          @jax.jit
          def g(y):
              return jnp.sin(y)
  
          return g(x)
  
      hlo_text = _get_hlo(f)(2.0)
      self.assertIn("jvp(jit(f))/jit(g)/sin", hlo_text)
      self.assertIn("jvp(jit(f))/jit(g)/cos", hlo_text)
      self.assertIn("transpose(jvp(jit(f)))/jit(g)/mul", hlo_text)
  
imports: import functools
  from absl.testing import absltest
  import jax
  import jax.numpy
  from jax._src import core
  from jax import lax
  from jax._src.pjit import pjit
  from jax._src import linear_util
  from jax import config
  from jax._src import test_util
  from jax._src.lib import xla_client
label_desc: Use image node_id 8 for calling the NameStackTransformationTest obj's underlying member method code with example usage: obj.test_nested_jit_stack() and returns: g, jnp