from typing import Any from torch.types import _bool # Defined in torch/csrc/cpu/Module.cpp def _init_amx() -> _bool: ... def _get_cpu_capability() -> dict[str, Any]: ...
