code: def __call__(self, x):
      return x @ self.w + self.b
  
file_path: flax/flax/experimental/nnx/examples/toy_examples/02_lifted_transforms.py
imports: import jax
  import jax.numpy
  import matplotlib.pyplot
  import numpy
  from flax.experimental import nnx
label_desc: Use image node_id 2 for calling the Linear obj's underlying member method code with example usage: obj.__call__(x) and returns: unknown