class_docstr: JAX tracer used in DotTrace.
class_docstr_tok: ["JAX","tracer","used","in","DotTrace","."]
code: def aval(self):
      return jax.core.get_aval(self.val)
  
imports: import collections
  import contextlib
  import functools
  import html
  from typing import Any, Callable, NamedTuple, Optional
  from haiku._src import data_structures
  from haiku._src import module
  from haiku._src import utils
  import jax
  import jax.core
  from jax.experimental import pjit
  from jax.extend import linear_util
label_desc: Use image node_id 2 for calling the DotTracer obj's underlying member method code with example usage: obj.aval() and returns: jax