Merge branch 'main' into lint_fixes#1_torchao/dtype

Author: jainapurva

torchao/dtypes/affine_quantized_tensor.py

@@ -1,6 +1,7 @@
 import torch
 from typing import Tuple, Optional, Union
-import torchao.ops
+from collections import defaultdict
+import functools
 import math
 from torchao.quantization.quant_primitives import (
     choose_qparams_affine,