feat(pt): DPLR in PyTorch backend#5138

Open

ChiahsinChu wants to merge 15 commits intodeepmodeling:masterfrom

ChiahsinChu:devel_pt_dplr

Commits on Jan 16, 2026

add dplr(torch)
ChiahsinChu
committed
update torch_admp version; minor update in dipole_charge non-pbc assertion
ChiahsinChu
committed
minor code improvement based on @coderabbitai
ChiahsinChu
committed
change source of torch_admp from github to pypi in pyproject.toml; minor code improvement
ChiahsinChu
committed
add torch-admp to dependencies

njzjz
authored and
ChiahsinChu
committed
pass tuple

njzjz
authored and
ChiahsinChu
committed
Fix pop method for torch_static_requirement

njzjz
authored and
ChiahsinChu
committed
docs: add PyTorch backend documentation to DPLR model
ChiahsinChu
committed
update freeze/eval with model (de-)serialization
ChiahsinChu
committed
Feat: Refactor DipoleChargeModifier to support direct model object input and improve serialization. Added support for both model file path and direct model object initialization, along with proper …
ChiahsinChu
committed
feat: Add batch processing support to DipoleChargeModifier for improved memory efficiency and performance.
ChiahsinChu
committed
feat: add DipoleChargeModifier.eval_np
ChiahsinChu
committed

Commits on Jan 17, 2026