diff --git a/onnxscript/__init__.py b/onnxscript/__init__.py
index bee5a1b23..96f1fa5ef 100644
--- a/onnxscript/__init__.py
+++ b/onnxscript/__init__.py
@@ -58,6 +58,7 @@
     "opset_ai_onnx_ml2",
     "opset_ai_onnx_ml3",
     "opset_ai_onnx_ml4",
+    "DEBUG",
 ]
 
 import importlib.metadata
@@ -122,6 +123,9 @@
 from ._internal.utils import external_tensor
 from .values import OnnxFunction, TracedOnnxFunction
 
+# Set DEBUG to True to enable additional debug checks
+DEBUG = False
+
 try:  # noqa: SIM105
     __version__ = importlib.metadata.version("onnxscript")
 except importlib.metadata.PackageNotFoundError: