diff --git a/imperative/python/megengine/autodiff/__init__.py b/imperative/python/megengine/autodiff/__init__.py
index 3def6f814714ee909a5ae722b87c99ae30b23633..10553b392b46b1e40c38e144a929c1112ae33896 100644
--- a/imperative/python/megengine/autodiff/__init__.py
+++ b/imperative/python/megengine/autodiff/__init__.py
@@ -6,4 +6,5 @@
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+from ..core.autodiff.grad import Function
 from .grad_manager import GradManager
diff --git a/imperative/python/megengine/quantization/fake_quant.py b/imperative/python/megengine/quantization/fake_quant.py
index 66bc56b6da71c1686709625d07d9653223d464ad..6b035d5316e0ffc2431494de6883d65364e9a611 100644
--- a/imperative/python/megengine/quantization/fake_quant.py
+++ b/imperative/python/megengine/quantization/fake_quant.py
@@ -11,7 +11,7 @@ from typing import Iterable
 import numpy as np
 
 from .. import functional as F
-from ..core.autodiff.grad import Function
+from ..autodiff import Function
 from ..core.tensor.dtype import _metadata_dict, get_quantized_dtype
 from ..module import Module
 from ..tensor import Parameter, Tensor
diff --git a/imperative/python/megengine/quantization/internal_fake_quant.py b/imperative/python/megengine/quantization/internal_fake_quant.py
index ab980406eed865239213d696b8e216c9bda8b1ce..1f246f4821ac5b89e4ee2c6c46130fb4e52ac579 100644
--- a/imperative/python/megengine/quantization/internal_fake_quant.py
+++ b/imperative/python/megengine/quantization/internal_fake_quant.py
@@ -12,7 +12,7 @@ from functools import partial
 import numpy as np
 
 from .. import functional as F
-from ..core.autodiff.grad import Function
+from ..autodiff import Function
 from .fake_quant import _FakeQuantize
 from .observer import MinMaxObserver
 from .qconfig import QConfig
diff --git a/imperative/python/megengine/quantization/utils.py b/imperative/python/megengine/quantization/utils.py
index 3c297cf7f9c1efea26ae3fc018b92cf418566c16..7f36c25b3264e10b8cc8a2bef554d0384b0a0c8f 100644
--- a/imperative/python/megengine/quantization/utils.py
+++ b/imperative/python/megengine/quantization/utils.py
@@ -12,8 +12,8 @@ from typing import Dict
 import numpy as np
 
 from .. import functional as F
+from ..autodiff import Function
 from ..core._imperative_rt.core2 import apply
-from ..core.autodiff.grad import Function
 from ..core.ops import builtin
 from ..core.tensor import megbrain_graph
 from ..core.tensor.dtype import _metadata_dict
diff --git a/imperative/python/test/unit/core/test_function.py b/imperative/python/test/unit/core/test_function.py
index accc751067e25eb62fbf40e39eda431d8a66deea..de4094323ecd8f662130e3376ce393a09e974c12 100644
--- a/imperative/python/test/unit/core/test_function.py
+++ b/imperative/python/test/unit/core/test_function.py
@@ -15,7 +15,7 @@ import megengine.optimizer as optimizer
 from megengine import Parameter
 from megengine import Tensor as tensor
 from megengine import tensor
-from megengine.core.autodiff.grad import Function
+from megengine.autodiff import Function
 from megengine.module import Module