class_docstr: Holds per-thread state on mixed precision policies.
class_docstr_tok: ["Holds","per-thread","state","on","mixed","precision","policies","."]
code: def __init__(self):
      super().__init__()
      self._installed_interceptor = False
      self._cls_policy: dict[ClassInfoOrType, jmp.Policy] = {}
      self._current_policy = Stack[jmp.Policy]()
  
imports: import collections
  import contextlib
  import threading
  from typing import TypeVar, Optional, Union
  from haiku._src import base
  from haiku._src import data_structures
  from haiku._src import module
  import jmp
label_desc: Use image node_id 1 to create a new _ThreadState object from inherited base classes: threading with example: obj = _ThreadState()