feat(tf2): add eager TensorFlow array backend

.github/workflows/test_python.yml

@@ -63,12 +63,36 @@ jobs:
           DP_CI_IMPORT_PADDLE_BEFORE_TF: 1
           FLAGS_use_stride_compute_kernel: 0
       - name: Test TF2 eager mode
-        run: pytest --cov=deepmd --cov-append source/tests/consistent/io/test_io.py source/jax2tf_tests
+        run: |
+          run_pytest_allow_no_tests() {
+            set +e
+            pytest "$@"
+            local status=$?
+            set -e
+            if [ "$status" -eq 5 ]; then
+              # pytest-split may leave an individual shard with no selected
+              # tests after path/-k filtering. Other shards still cover the
+              # selected tests, so do not fail the whole matrix for exit 5.
+              return 0
+            fi
+            return "$status"
+          }
+
+          run_pytest_allow_no_tests --cov=deepmd --cov-append \
+            source/tests/consistent/io/test_io.py \
+            source/jax2tf_tests \
+            --splits 12 \
+            --group ${{ matrix.group }}
+          run_pytest_allow_no_tests --cov=deepmd --cov-append \
+            source/tests/consistent \
+            -k tf2 \
+            --splits 12 \
+            --group ${{ matrix.group }}
         env:
           NUM_WORKERS: 0
           DP_TEST_TF2_ONLY: 1
           DP_DTYPE_PROMOTION_STRICT: 1
-        if: matrix.group == 1
+          DP_CI_IMPORT_PADDLE_BEFORE_TF: 1
       - run: mv .test_durations .test_durations_${{ matrix.group }}
       - name: Upload partial durations
         uses: actions/upload-artifact@v7

deepmd/_vendors/__init__.py

@@ -0,0 +1,2 @@
+# SPDX-License-Identifier: LGPL-3.0-or-later
+"""Vendored third-party modules used by DeePMD-kit."""

deepmd/_vendors/ndtensorflow/__init__.py

@@ -0,0 +1,38 @@
+# SPDX-License-Identifier: LGPL-3.0-or-later
+from __future__ import (
+    annotations,
+)
+
+from typing import (
+    Final,
+)
+
+from . import (
+    fft,
+    linalg,
+)
+from ._array import (
+    Array,
+)
+from ._info import (
+    __array_namespace_info__,
+)
+from ._namespace import *
+from ._namespace import __all__ as _namespace_all
+
+__array_api_version__: Final = "2025.12"
+
+__all__ = sorted(
+    set(_namespace_all)
+    | {
+        "Array",
+        "__array_api_version__",
+        "__array_namespace_info__",
+        "fft",
+        "linalg",
+    }
+)
+
+
+def __dir__() -> list[str]:
+    return __all__