class_docstr: Provides friendlier syntax for mapping with `jax.tree_util.tree_map`.
  
  !!! example
  
      ```python
      (ω(a) + ω(b)).ω == jax.tree_util.tree_map(operator.add, a, b)
      ```
  
  !!! tip
  
      To minimise the number of brackets, the following `__rpow__` syntax can be
      used:
  
      ```python
      (a**ω + b**ω).ω == jax.tree_util.tree_map(operator.add, a, b)
      ```
  
      This is entirely equivalent to the above.
code: def __repr__(self):
      return f"ω({self.ω})"
  
imports: import operator
  from collections.abc import Callable
  from typing import Any, Optional, TYPE_CHECKING
  import jax
  import jax.tree_util
label_desc: Use image node_id 2 for calling the _ω obj's underlying member method code with example usage: obj.__repr__() and returns: str, self, str