feat(pt_expt): DPA4 descriptor and fitting (training + export) - #5522

Merged

wanghan-iapcm merged 12 commits into

deepmodeling:masterfrom

wanghan-iapcm:feat-ptexpt-dpa4

Jun 15, 2026

Commits on Jun 12, 2026

feat(pt_expt): DPA4 descriptor wrapper with converter registrations
Han Wang
committed
3bdc8f9
feat(pt_expt): dpa4_ener fitting wrapper
Han Wang
committed
6acc0ef
feat(pt_expt): dpa4 model-type assembly
Han Wang
committed
50e6a7d
fix(pt_expt): guard preset_out_bias, document enable_tf32 in dpa4 assembly
Han Wang
committed
b3651b9
test(pt_expt): tighten dpa4 assembly guard assertions
Han Wang
committed
c8cd878
test(pt_expt): DPA4 training end-to-end
Han Wang
committed
c0bae02
fix(pt_expt): promote DPA4 trainable weights to Parameters; grad parity vs pt
Han Wang
committed
7aca880
test(consistent): enable pt_expt rows for DPA4
Han Wang
committed
1f26ba1
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
b1920ee

Commits on Jun 13, 2026

fix(pt_expt): address review findings on DPA4 PR (trainable=False freeze, component type validation)
Han Wang
committed
3749300

Commits on Jun 14, 2026

fix(dpa4): convert rebuilt index buffers via to_numpy_array in deserialize (CUDA)
Han Wang
committed
79bce06
fix(dpa4): convert mean/stddev/wigner buffers via to_numpy_array in serialize (CUDA)
Han Wang
committed
56793cf