Commit 774a844
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- deepmd
- dpmodel
- descriptor
- utils
- jax/descriptor
- pd/utils
- pt
- model
- descriptor
- task
- utils
- tf
- utils
- source/tests
- array_api_strict/descriptor
- consistent/descriptor
- pt
- model
- universal
- dpmodel/descriptor
- pt
- descriptor
- model
27 files changed
+5308
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
53 | 53 | | |
54 | 54 | | |
55 | 55 | | |
| 56 | + | |
56 | 57 | | |
57 | 58 | | |
58 | 59 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
5 | 5 | | |
6 | 6 | | |
7 | 7 | | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
8 | 11 | | |
9 | 12 | | |
10 | 13 | | |
| |||
30 | 33 | | |
31 | 34 | | |
32 | 35 | | |
| 36 | + | |
33 | 37 | | |
34 | 38 | | |
35 | 39 | | |
| |||
0 commit comments