diff --git a/onnxscript/__init__.py b/onnxscript/__init__.py
index 9783e37c24..6731631aae 100644
--- a/onnxscript/__init__.py
+++ b/onnxscript/__init__.py
@@ -13,6 +13,7 @@
     "OnnxFunction",
     "TracedOnnxFunction",
     "GraphBuilder",
+    "OpBuilder",
     "proto2python",
     "external_tensor",
     "BFLOAT16",
@@ -129,7 +130,7 @@
 # isort: on
 
 from . import ir, nn, optimizer, rewriter, version_converter
-from ._internal.builder import GraphBuilder
+from ._internal.builder import GraphBuilder, OpBuilder
 from ._internal.utils import external_tensor
 from ._internal.values import OnnxFunction, TracedOnnxFunction