code: def test_lift_multiple_uses(self):
      def f(x):
          mod = transform_flax.lift(nn.Dense(1), name="mod")
          x = mod(x)
          x = mod(x)
          return x
  
      f = transform.transform(f)
      rng = jax.random.PRNGKey(42)
      x = jnp.ones([1, 1])
      params = f.init(rng, x)
      params = jax.tree_map(lambda x: (x.shape, x.dtype), params)
      self.assertEqual(
          params,
          {
              "mod/~": {
                  "bias": ((1,), jnp.float32),
                  "kernel": ((1, 1), jnp.float32),
              }
          },
      )
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import flax.errors
  import flax.linen
  from haiku._src import base
  from haiku._src import module
  from haiku._src import transform
  from haiku._src.flax import transform_flax
  import jax
  import jax.numpy
label_desc: Use image node_id 11 for calling the TransformFlaxTest obj's underlying member method code with example usage: obj.test_lift_multiple_uses() and returns: x