class_docstr: Unit tests for variable tracing using.
code: def test_lambda_inside_function(self):
      m = objax.nn.Sequential(
          [objax.nn.Conv2D(1, 2, 3), objax.nn.BatchNorm2D(2)]
      )
  
      def loss(x):
          get_logits = lambda inp: m(inp, training=True)
          return jn.sum(get_logits(x))
  
      vc = objax.util.find_used_variables(loss)
      self.assertDictEqual(vc, m.vars(scope="m."))
  
file_path: objax/tests/tracing.py
imports: import unittest
  import numpy
  import jax.numpy
  import objax
  from objax.zoo.dnnet import DNNet
label_desc: Use image node_id 9 for calling the TestTracing obj's underlying member method code with example usage: obj.test_lambda_inside_function() and returns: jn