diff --git a/source/checker/deepmd_checker.py b/source/checker/deepmd_checker.py
index d763835fdc..9a1538f1e3 100644
--- a/source/checker/deepmd_checker.py
+++ b/source/checker/deepmd_checker.py
@@ -37,7 +37,7 @@ def visit_call(self, node):
         if (
             isinstance(node.func, Attribute)
             and isinstance(node.func.expr, Name)
-            and node.func.expr.name in {"np", "tf", "torch"}
+            and node.func.expr.name in {"np", "tf", "torch", "xp", "jnp}
             and node.func.attrname
             in {
                 # https://pytorch.org/docs/stable/torch.html#creation-ops