You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
| 🧬 |**Pretrained-first workflows**| Download [pretrained DPA models][pretrained], fine-tune full models or LoRA adapters, or adapt learned representations to downstream properties with [DPA-ADAPT].|
44
-
| 🏗️ |**Training from scratch**| Design a model for a new system or physical target, then train it with single-task, multi-task, and distributed workflows across supported backends. |
45
-
| 🧠 |**Modern model portfolio**| Start with efficient DeepPot-SE descriptors or move to [DPA][model-guide] for large atomistic models. |
46
-
| 🧲 |**More than energy and force**| Model virials, Hessians, spin and magnetic forces, dipoles, polarizabilities, electronic density of states, atomic populations, and arbitrary intensive or extensive properties. |
47
-
| 🔄 |**Backend flexibility**| Train or run supported models with [TensorFlow, PyTorch, JAX, or Paddle][backends], with backend-aware model formats and conversion paths for compatible architectures. |
48
-
| 🚀 |**Performance from training to MD**| Use CPUs, CUDA GPUs, ROCm source builds, distributed training, model compression, compiled DPA-4 paths, AOTInductor `.pt2` export, and MPI-enabled simulation. |
49
-
| 🔌 |**Deploy where science happens**| Use the CLI, Python, C, C++, or Node.js, then connect models to LAMMPS, i-PI, ASE, GROMACS, JAX MD, nvalchemi, OpenMM, Amber, CP2K, ABACUS, and more. |
50
-
| 🧩 |**Open and extensible**| Compose hybrid potentials, add analytical ZBL or long-range corrections, create custom models and operators, or connect external GNNs such as MACE and NequIP through plugins. |
| 🧬 |**Pretrained-first workflows**| Download [pretrained DPA models][pretrained], fine-tune full models, use [DPA-4 LoRA adapters][dpa4-lora] with PyTorch single-task training, or adapt learned representations to downstream properties with [DPA-ADAPT]. |
44
+
| 🏗️ |**Training from scratch**| Design a model for a new system or physical target, then train it with single-task, multi-task, and distributed workflows across supported backends. |
45
+
| 🧠 |**Modern model portfolio**| Start with efficient DeepPot-SE descriptors or move to [DPA][model-guide] for large atomistic models. |
46
+
| 🧲 |**More than energy and force**| Model virials, Hessians, spin and magnetic forces, dipoles, polarizabilities, electronic density of states, atomic populations, and arbitrary intensive or extensive properties. |
47
+
| 🔄 |**Backend flexibility**| Train or run supported models with [TensorFlow, PyTorch, JAX, or Paddle][backends], with backend-aware model formats and conversion paths for compatible architectures. |
48
+
| 🚀 |**Performance from training to MD**| Use CPUs, CUDA GPUs, ROCm source builds, distributed training, model compression, compiled DPA-4 paths, AOTInductor `.pt2` export, and MPI-enabled simulation. |
49
+
| 🔌 |**Deploy where science happens**| Use the CLI, Python, C, C++, or Node.js, then connect models to LAMMPS, i-PI, ASE, GROMACS, JAX MD, nvalchemi, OpenMM, Amber, CP2K, ABACUS, and more. |
50
+
| 🧩 |**Open and extensible**| Compose hybrid potentials, add analytical ZBL or long-range corrections, create custom models and operators, or connect external GNNs such as MACE and NequIP through plugins. |
51
51
52
52
> [!TIP]
53
53
> On supported descriptors and workloads, [model compression][compression] can
@@ -76,9 +76,10 @@ flowchart LR
76
76
adaptation, or configure a model to train from scratch.
77
77
1.**Prepare target data** in DeePMD's NumPy format or convert structures and
78
78
trajectories with [dpdata][data].
79
-
1.**Fine-tune or train:** adapt the full pretrained model or LoRA adapters,
80
-
or optimize a new model with single-task, multi-task, and distributed
81
-
training workflows.
79
+
1.**Fine-tune or train:** adapt the full pretrained model, use
80
+
[DPA-4 LoRA adapters][dpa4-lora] with PyTorch single-task training, or
81
+
optimize a new model with single-task, multi-task, and distributed training
82
+
workflows.
82
83
1.**Validate and export** with [`dp test`][testing], [`dp freeze`][freeze],
83
84
backend conversion, embedding extraction, and supported compression paths.
84
85
1.**Run simulation** through Python or native APIs, or load the model into a
@@ -119,7 +120,8 @@ mode with `dp --pt` and select the branch that matches your system with
119
120
configuration, so `input.json` does not need to reproduce the DPA-3.2
120
121
architecture.
121
122
122
-
The [fine-tuning guide][finetune] covers full-model and LoRA adaptation.
123
+
The [fine-tuning guide][finetune] covers full-model adaptation. [DPA-4 LoRA
124
+
fine-tuning][dpa4-lora] is available for PyTorch single-task training.
123
125
[DPA-ADAPT] reuses pretrained DPA representations for downstream
124
126
property-prediction tasks.
125
127
@@ -256,6 +258,7 @@ DeePMD-kit is licensed under the
0 commit comments