code: def __init__(self, keys, values):
      assert len(keys) == len(values)
      self._dict = {id(k): v for k, v in zip(keys, values)}
      self._count = {id(k): 0 for k in keys}
  
imports: from collections.abc import Callable, Sequence
  from typing import Any, Optional, TYPE_CHECKING, Union
  import jax.numpy
  import jax.tree_util
  import numpy
  from jaxtyping import Array, ArrayLike, Bool, Float, PyTree, PyTreeDef
  from ._custom_types import sentinel
  from ._doc_utils import doc_repr
  from ._filters import is_array
label_desc: Use image node_id 1 to create a new _CountedIdDict object  with example: obj = _CountedIdDict(keys, values)