feat(dpmodel/jax): add entry point for dpmodel and jax backend (#4284)

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **New Features**
- Introduced entry point loading functionality for enhanced module
initialization in both `dpmodel` and `jax` components of the DeepMD
framework.

These changes improve the framework's functionality and streamline
backend configuration.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->

Signed-off-by: Jinzhe Zeng <[email protected]>

Author: njzjz

deepmd/dpmodel/__init__.py

@@ -1,4 +1,8 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
+from deepmd.utils.entry_point import (
+    load_entry_point,
+)
+
 from .common import (
     DEFAULT_PRECISION,
     PRECISION_DICT,
@@ -32,3 +36,6 @@
     "get_deriv_name",
     "get_hessian_name",
 ]
+
+
+load_entry_point("deepmd.dpmodel")

deepmd/jax/__init__.py

@@ -1,2 +1,8 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 """JAX backend."""
+
+from deepmd.utils.entry_point import (
+    load_entry_point,
+)
+
+load_entry_point("deepmd.jax")