import torch.ao.quantization.pt2e raises ModuleNotFoundError in Torch 2.11 as it was moved to torchao. So from quark.torch.kernel import mx ultimately fails when trying from torch.ao.quantization.pt2e.utils import _get_tensor_constant_from_node
ModuleNotFoundError: No module named 'torch.ao.quantization.pt2e'
import torch.ao.quantization.pt2eraisesModuleNotFoundErrorin Torch 2.11 as it was moved totorchao. Sofrom quark.torch.kernel import mxultimately fails when tryingfrom torch.ao.quantization.pt2e.utils import _get_tensor_constant_from_nodeModuleNotFoundError: No module named 'torch.ao.quantization.pt2e'