diff --git a/onnxscript/__init__.py b/onnxscript/__init__.py
index 5b745807d3..829ee212cf 100644
--- a/onnxscript/__init__.py
+++ b/onnxscript/__init__.py
@@ -16,6 +16,8 @@
     "OpBuilder",
     "OpBuilderBase",
     "TapeBuilder",
+    "build_function",
+    "build_graph",
     "proto2python",
     "external_tensor",
     "BFLOAT16",
@@ -132,7 +134,7 @@
 # isort: on
 
 from . import ir, nn, optimizer, rewriter, version_converter
-from ._internal.builder import GraphBuilder, OpBuilder
+from ._internal.builder import GraphBuilder, OpBuilder, build_function, build_graph
 from ._internal.tape_builder import OpBuilderBase, TapeBuilder
 from ._internal.utils import external_tensor
 from ._internal.values import OnnxFunction, TracedOnnxFunction