code: def test_pytree_with_new(self):
      class A(nnx.Pytree):
          def __init__(self, a):
              self.a = a
  
          def __new__(cls, a):
              return super().__new__(cls)
  
      pytree = A(a=1)
  
      pytree = jax.tree_map(lambda x: x * 2, pytree)
  
file_path: flax/flax/experimental/nnx/tests/test_pytree.py
imports: from typing import Generic, TypeVar
  import jax
  import pytest
  from flax import serialization
  from flax.experimental import nnx
label_desc: Use image node_id 9 for calling the TestPytree obj's underlying member method code with example usage: obj.test_pytree_with_new() and returns: super