diff --git a/torchao/prototype/quantization/codebook_groupwise/__init__.py b/torchao/prototype/quantization/codebook_groupwise/__init__.py
new file mode 100644
index 0000000000..26076581e1
--- /dev/null
+++ b/torchao/prototype/quantization/codebook_groupwise/__init__.py
@@ -0,0 +1,4 @@
+from .api import GroupwiseLutWeightConfig
+from .codebook_quantized_tensor import GroupwiseLutQuantizedTensor
+
+__all__ = ["GroupwiseLutQuantizedTensor", "GroupwiseLutWeightConfig"]