@@ -20,28 +20,28 @@ DeePMD-kit v3 implements a flexible and pluggable backend framework, providing a
2020
2121| Function | TensorFlow | PyTorch | JAX | DP |
2222| ------------------------------| ------------| ---------| ------| ------|
23- | Local frame descriptor | ✅ | ✅ | ✅ | ✅ |
23+ | Local frame descriptor | ✅ | | | |
2424| se_e2_a descriptor | ✅ | ✅ | ✅ | ✅ |
2525| se_e2_r descriptor | ✅ | ✅ | ✅ | ✅ |
2626| se_e3 descriptor | ✅ | ✅ | ✅ | ✅ |
27- | se_e3_tebd descriptor | ✅ | ✅ | ✅ | ✅ |
27+ | se_e3_tebd descriptor | | ✅ | ✅ | ✅ |
2828| DPA-1 descriptor | ✅ | ✅ | ✅ | ✅ |
29- | DPA-2 descriptor | ✅ | ✅ | ✅ | ✅ |
29+ | DPA-2 descriptor | | ✅ | ✅ | ✅ |
3030| Hybrid descriptor | ✅ | ✅ | ✅ | ✅ |
3131| Fit energy | ✅ | ✅ | ✅ | ✅ |
3232| Fit dipole | ✅ | ✅ | ✅ | ✅ |
3333| Fit polar | ✅ | ✅ | ✅ | ✅ |
3434| Fit DOS | ✅ | ✅ | ✅ | ✅ |
35- | Fit properties | ✅ | ✅ | ✅ | ✅ |
35+ | Fit properties | | ✅ | ✅ | ✅ |
3636| ZBL | ✅ | ✅ | ✅ | ✅ |
37- | DPLR | ✅ | ✅ | ✅ | ✅ |
37+ | DPLR | ✅ | | | |
3838| DPRc | ✅ | ✅ | ✅ | ✅ |
39- | Spin | ✅ | ✅ | ✅ | ✅ |
40- | Ladder calculation | ✅ | ✅ | ✅ | ✅ |
41- | Model training | ✅ | ✅ | ✅ | ✅ |
42- | Model compression | ✅ | ✅ | ✅ | ✅ |
39+ | Spin | ✅ | ✅ | | ✅ |
40+ | Ladder calculation | ✅ | ✅ | ✅ | |
41+ | Model training | ✅ | ✅ | | |
42+ | Model compression | ✅ | ✅ | | |
4343| Python inference | ✅ | ✅ | ✅ | ✅ |
44- | C++ inference | ✅ | ✅ | ✅ | ✅ |
44+ | C++ inference | ✅ | ✅ | ✅ | |
4545
4646
4747The main features of the multi-backend framework include:
0 commit comments