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 __getitem__(self, item):
      return ω(
          jtu.tree_map(lambda x: x[item], self.ω, is_leaf=self.is_leaf),
          is_leaf=self.is_leaf,
      )
  
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 3 for calling the _ω obj's underlying member method code with example usage: obj.__getitem__(item) and returns: ω