diff --git a/internlm/initialize/__init__.py b/internlm/initialize/__init__.py
index ae94e0a..14fe06b 100644
--- a/internlm/initialize/__init__.py
+++ b/internlm/initialize/__init__.py
@@ -4,6 +4,7 @@ from .launch import (
     initialize_distributed_env,
     launch_from_slurm,
     launch_from_torch,
+    try_bind_numa,
 )
 
 __all__ = [
@@ -12,4 +13,5 @@ __all__ = [
     "launch_from_slurm",
     "launch_from_torch",
     "initialize_distributed_env",
+    "try_bind_numa",
 ]