Skip to content

Commit 774a844

Browse files
authored
feat(pt/dp/jax/xp): add DPA3 descriptor (#4609)
**NOTE**: examples, custom ops and docs will be updated in other PRs. <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **New Features** - Introduced the `DescrptDPA3` descriptor for advanced molecular simulations, enhancing flexibility and performance. - Added support for the SiLU activation function, expanding options for neural network configurations. - Integrated refined repflow modules to improve message passing and descriptor computations. - Added a new model configuration `model_dpa3` for testing purposes. - **Tests** - Expanded test coverage to include the new `DescrptDPA3` descriptor and repflow functionalities, ensuring consistency and reliability across different backends. - Implemented comprehensive unit tests for `DescrptDPA3` to validate its functionality and performance across various configurations. - Added parameterized tests for `DescrptDPA3` to assess its behavior under different conditions. - Introduced a new test class `TestMultiTaskDPA3` for multitask training scenarios involving `model_dpa3`. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
1 parent f1f1b7b commit 774a844

File tree

27 files changed

+5308
-1
lines changed

27 files changed

+5308
-1
lines changed

deepmd/common.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@
5353
"tanh",
5454
"gelu",
5555
"gelu_tf",
56+
"silu",
5657
"none",
5758
"linear",
5859
]

deepmd/dpmodel/descriptor/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,9 @@
55
from .dpa2 import (
66
DescrptDPA2,
77
)
8+
from .dpa3 import (
9+
DescrptDPA3,
10+
)
811
from .hybrid import (
912
DescrptHybrid,
1013
)
@@ -30,6 +33,7 @@
3033
__all__ = [
3134
"DescrptDPA1",
3235
"DescrptDPA2",
36+
"DescrptDPA3",
3337
"DescrptHybrid",
3438
"DescrptSeA",
3539
"DescrptSeAttenV2",

0 commit comments

Comments
 (0)