Add AOPerModuleConfig to torchao.quantization (#2134)

* Add AOPerModuleConfig to `torchao.quantization`

* Apply automatic Ruff fixes

---------

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

Author: 2 people

torchao/quantization/__init__.py

@@ -39,6 +39,7 @@
     AffineQuantizedObserverBase,
 )
 from .quant_api import (
+    AOPerModuleConfig,
     CutlassInt4PackedLayout,
     Float8DynamicActivationFloat8SemiSparseWeightConfig,
     Float8DynamicActivationFloat8WeightConfig,
@@ -140,6 +141,7 @@
     "UIntXWeightOnlyConfig",
     "FPXWeightOnlyConfig",
     "GemliteUIntXWeightOnlyConfig",
+    "AOPerModuleConfig",
     # smooth quant - subject to change
     "get_scale",
     "SmoothFakeDynQuantMixin",